我一直在跟随这个教程来理解线性分类模型及其应用。我在人口普查数据之外举了一个不同的例子,我可以用evaluate得到准确度。在
现在我感兴趣的是用预测的列值打印出测试数据行。在import random
import pandas
import tensorflow as tf
import tempfile
import numpy as np
df_train = pandas.read_csv('input/train.csv', usecols=['Sex', 'Age', 'Fare','Survived', 'SibSp'])
df_test = pandas.read_csv('input/test.csv', usecols=['Sex', 'Age', 'Fare', 'SibSp'])
df_test['Survived'] = 0
CATEGORICAL_COLUMNS = ['Sex']
CONTINUOUS_COLUMNS = ['Age', 'Fare', 'SibSp']
df_train_nona = df_train.dropna()
df_test_nona = df_test.dropna()
print(df_test_nona)
def input_fn(df):
continuous_cols = {k: tf.constant(df[k].values)
for k in CONTINUOUS_COLUMNS}
<