Merge pull request #2 from Isdre/test

update requirements.txt
Isdre · Jun 14, 2024 · b887cd0 · b887cd0
2 parents 1695f68 + 5a0ea2c
commit b887cd0
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/.idea/neuralNetwork.iml b/.idea/neuralNetwork.iml
diff --git a/requirements.txt b/requirements.txt
diff --git a/test_compare_tensorflow.py b/test_compare_tensorflow.py
@@ -20,7 +20,7 @@ def test_digit_tensorflow():
     network.add(Layer(neurons_count=10))
     network.compile(loss=SquareError,optimizer=Adam(0.0002),metric=Accuracy)
 
-    network.fit(X_train,y_train,epochs=3,batch_size=1, validation_split=0.1)
+    network.fit(X_train,y_train,epochs=3,batch_size=32, validation_split=0.1)
 
     model = tf.keras.models.Sequential([
         tf.keras.layers.Flatten(input_shape=(28, 28)),