Saving and running wide_deep.py model

Viewed 602

I've been playing around with the Tensorflow Wide and Deep tutorial using the census dataset.

The linear/wide tutorial states:

We will train a logistic regression model, and given an individual's information our model will output a number between 0 and 1

At the moment, I can't work out how to predict the output of an individual input (copied from the unit test):

TEST_INPUT_VALUES = {
    'age': 18,
    'education_num': 12,
    'capital_gain': 34,
    'capital_loss': 56,
    'hours_per_week': 78,
    'education': 'Bachelors',
    'marital_status': 'Married-civ-spouse',
    'relationship': 'Husband',
    'workclass': 'Self-emp-not-inc',
    'occupation': 'abc',
}

How can we predict and output whether this person is likely to earn <50k (0) or >=50k (1)?

2 Answers
Related