Update machine_learning.py
Browse files- machine_learning.py +2 -2
machine_learning.py
CHANGED
@@ -173,8 +173,8 @@ output = numpy.array(output)
|
|
173 |
tensorflow.compat.v1.reset_default_graph()
|
174 |
|
175 |
net = tflearn.input_data(shape=[None, len(training[0])])
|
176 |
-
net = tflearn.fully_connected(net,
|
177 |
-
net = tflearn.fully_connected(net,
|
178 |
net = tflearn.fully_connected(net, len(output[0]), activation="softmax")
|
179 |
net = tflearn.regression(net)
|
180 |
|
|
|
173 |
tensorflow.compat.v1.reset_default_graph()
|
174 |
|
175 |
net = tflearn.input_data(shape=[None, len(training[0])])
|
176 |
+
net = tflearn.fully_connected(net, 32)
|
177 |
+
net = tflearn.fully_connected(net, 32)
|
178 |
net = tflearn.fully_connected(net, len(output[0]), activation="softmax")
|
179 |
net = tflearn.regression(net)
|
180 |
|