Lines Matching refs:train
59 if mode == tf.estimator.ModeKeys.TRAIN:
60 optimizer = params.get("optimizer", tf.train.AdamOptimizer)
63 loss=average_loss, global_step=tf.train.get_global_step())
87 (train, test) = imports85.dataset()
93 train = train.map(normalize_price)
101 train.shuffle(1000).batch(128)
141 "optimizer": tf.train.AdamOptimizer,
145 # Train the model.
146 model.train(input_fn=input_train, steps=STEPS)
