1

テンソルボードでの各反復後に、この DNNRegression モデルの精度を示す方法はありますか? 私が見た唯一の方法は、tf.estimator を使用するのではなく、「セッション」メソッドを使用することです。また、手作業に頼らずにモデルの最終的な精度を見つける方法はありますか? 評価方法を試してみましたが、返される辞書に「accuracy」キーがありません。

import numpy as np
import tensorflow as tf
import _pickle as cPickle

with open("var_x.txt", "rb") as fp:   # Unpickling
    var_x = cPickle.load(fp)

with open("var_y.txt", "rb") as fp:   # Unpickling
    var_y = cPickle.load(fp)

with open("var_x_test.txt", "rb") as fp:   # Unpickling
    var_x_test = cPickle.load(fp)

with open("var_y_test.txt", "rb") as fp:   # Unpickling
    var_y_test = cPickle.load(fp)

test_set = tf.contrib.learn.datasets.base.load_csv_with_header(
        filename="test.csv",
        target_dtype=np.float64,
        features_dtype=np.float64)

feature_columns = [tf.feature_column.numeric_column("x", shape=[4])]

estimator = tf.estimator.DNNRegressor(feature_columns=feature_columns, hidden_units=[1024, 512, 256])

# define our data sets
x_train = np.array(var_x)
y_train = np.array(var_y)
x_test = np.array(var_x_test)
y_test = np.array(var_y_test)

input_fn = tf.estimator.inputs.numpy_input_fn(
    {"x": x_train}, y_train, batch_size=4, num_epochs=60, shuffle=True)

# train
estimator.train(input_fn=input_fn, steps=1000)

#TESTING
prediction_input_fn= tf.estimator.inputs.numpy_input_fn(
    x ={"x":x_test},
    num_epochs=1,
    shuffle=False
    )

predictions = list(estimator.predict(input_fn=prediction_input_fn))

s=0

for i in range(len(predictions)):
    print(str(int(abs(round(predictions[i]['predictions'][0]))))+"\n")
    if (int(abs(round(predictions[i]['predictions'][0]))) == y_test[i]):
        s+=1

print(s)
4

1 に答える 1