Update machine_learning.py
Browse files- machine_learning.py +1 -1
machine_learning.py
CHANGED
@@ -185,5 +185,5 @@ if os.path.exists('minh103.meta'):
|
|
185 |
model.load('minh103')
|
186 |
else:
|
187 |
model = tflearn.DNN(net)
|
188 |
-
model.fit(training, output, n_epoch=
|
189 |
model.save('minh103')
|
|
|
185 |
model.load('minh103')
|
186 |
else:
|
187 |
model = tflearn.DNN(net)
|
188 |
+
model.fit(training, output, n_epoch=50000, batch_size=10, show_metric=True)
|
189 |
model.save('minh103')
|