num_epoch should be 1 for testing input

feranick · Oct 22, 2017 · 3fbb01f · 3fbb01f
1 parent fce18b8
commit 3fbb01f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/SpectraLearnPredict/SpectraLearnPredict/slp/__init__.py b/SpectraLearnPredict/SpectraLearnPredict/slp/__init__.py
@@ -5,7 +5,7 @@
 *
 * SpectraLearnPredict
 * Perform Machine Learning on Spectroscopy Data.
-* version: 20171021a
+* version: 20171022a
 *
 * Uses: Deep Neural Networks, TensorFlow, SVM, PCA, K-Means
 *

diff --git a/SpectraLearnPredict/SpectraLearnPredict/slp/slp_tf.py b/SpectraLearnPredict/SpectraLearnPredict/slp/slp_tf.py
@@ -199,7 +199,7 @@ def trainDNNTF2(A, Cl, A_test, Cl_test, Root):
     test_input_fn = tf.estimator.inputs.numpy_input_fn(
             x={"x": np.array(A_test)},
             y=np.array(Cl2_test),
-            num_epochs=None,
+            num_epochs=1,
             shuffle=dnntfDef.shuffleTest)
 
     validation_monitor = [skflow.monitors.ValidationMonitor(input_fn=test_input_fn,

diff --git a/SpectraLearnPredict/setup.py b/SpectraLearnPredict/setup.py
@@ -5,7 +5,7 @@
     packages=find_packages(),
     install_requires=['numpy', 'matplotlib', 'pandas', 'tensorflow', 'scikit-learn'],
     entry_points={'gui_scripts' : ['SpectraLearnPredict=SpectraLearnPredict.__main__:main']},
-    version='20171021a',
+    version='20171022a',
     description='Machine learning for spectral data',
     long_description= """ Machine learning for spectral data """,
     author='Nicola Ferralis',