Update machine_learning.py
Browse files- machine_learning.py +1 -1
machine_learning.py
CHANGED
@@ -143,5 +143,5 @@ if os.path.exists('minh103.meta'):
|
|
143 |
model.load('minh103')
|
144 |
else:
|
145 |
model = tflearn.DNN(net)
|
146 |
-
model.fit(training, output, n_epoch=
|
147 |
model.save('minh103')
|
|
|
143 |
model.load('minh103')
|
144 |
else:
|
145 |
model = tflearn.DNN(net)
|
146 |
+
model.fit(training, output, n_epoch=20000, batch_size=10, show_metric=True)
|
147 |
model.save('minh103')
|