Skip to content

Commit 69b97e0

Browse files
author
Ragav Venkatesan
authored
Removed eval methods.
1 parent a8f75d7 commit 69b97e0

File tree

1 file changed

+2
-11
lines changed

1 file changed

+2
-11
lines changed

under_development/tensorflow_iris_byom/iris_dnn_classifier.py

Lines changed: 2 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,7 @@
22
import numpy as np
33
import tensorflow as tf
44

5-
INPUT_TENSOR_NAME = 'x'
6-
5+
INPUT_TENSOR_NAME = 'inputs'
76

87
def estimator_fn(run_config, params):
98
feature_columns = [tf.feature_column.numeric_column(INPUT_TENSOR_NAME, shape=[4])]
@@ -12,22 +11,14 @@ def estimator_fn(run_config, params):
1211
n_classes=3,
1312
config=run_config)
1413

15-
1614
def serving_input_fn():
1715
feature_spec = {INPUT_TENSOR_NAME: tf.FixedLenFeature(dtype=tf.float32, shape=[4])}
1816
return tf.estimator.export.build_parsing_serving_input_receiver_fn(feature_spec)()
1917

20-
2118
def train_input_fn(training_dir, params):
2219
"""Returns input function that would feed the model during training"""
2320
return _generate_input_fn(training_dir, 'iris_training.csv')
2421

25-
26-
def eval_input_fn(training_dir, params):
27-
"""Returns input function that would feed the model during evaluation"""
28-
return _generate_input_fn(training_dir, 'iris_test.csv')
29-
30-
3122
def _generate_input_fn(training_dir, training_filename):
3223
training_set = tf.contrib.learn.datasets.base.load_csv_with_header(
3324
filename=os.path.join(training_dir, training_filename),
@@ -38,4 +29,4 @@ def _generate_input_fn(training_dir, training_filename):
3829
x={INPUT_TENSOR_NAME: np.array(training_set.data)},
3930
y=np.array(training_set.target),
4031
num_epochs=None,
41-
shuffle=True)
32+
shuffle=True)

0 commit comments

Comments
 (0)