plugin: model: tensorflow Change hidden_units

Signed-off-by: John Andersen <[email protected]>

Author: John Andersen

model/tensorflow/README.md

@@ -8,6 +8,10 @@ DFFML models backed by Tensorflow.
 
 ![Demo](https://github.com/intel/dffml/raw/master/docs/images/iris_demo.gif)
 
+> This demo was taken before the below steps were updated, the actual accuracy
+> may vary as this video shows accuracy being assessed against the training
+> data. You should try it for yourself and see!
+
 ## Install
 
 ```console
@@ -38,7 +42,7 @@ dffml train \
   -log debug
 dffml accuracy \
   -model dnn \
-  -sources csv=iris_training.csv \
+  -sources csv=iris_test.csv \
   -classifications 0 1 2 \
   -features \
     def:SepalLength:float:1 \

model/tensorflow/dffml_model_tensorflow/model/dnn.py

@@ -189,7 +189,7 @@ async def model(self, features: Features, classifications: List[Any]):
                 len(classifications), classifications)
         self._model = self._tf.estimator.DNNClassifier(
                 feature_columns=list((await self.features(features)).values()),
-                hidden_units=[10, 20, 10],
+                hidden_units=[12, 40, 15],
                 n_classes=len(classifications),
                 model_dir=self.model_dir_path(features))
         return self._model