Add requirements file

2019-05-06 14:45:47 +02:00 · 2019-05-06 14:45:47 +02:00 · 7f66120f25
commit 7f66120f25
parent 7b02c4f746
14 changed files with 8 additions and 2 deletions
--- a/.vscode/settings.json
+++ b/.vscode/settings.json
@ -0,0 +1,3 @@
+{
+    "python.pythonPath": "/usr/bin/python3.6"
+}
--- a/SZI1/pycache/init.cpython-36.pyc
+++ b/SZI1/pycache/init.cpython-36.pyc
--- a/SZI1/pycache/settings.cpython-36.pyc
+++ b/SZI1/pycache/settings.cpython-36.pyc
--- a/SZI1/pycache/urls.cpython-36.pyc
+++ b/SZI1/pycache/urls.cpython-36.pyc
--- a/SZI1/pycache/wsgi.cpython-36.pyc
+++ b/SZI1/pycache/wsgi.cpython-36.pyc
--- a/iris_model.h5
+++ b/iris_model.h5
--- a/magazine/pycache/init.cpython-36.pyc
+++ b/magazine/pycache/init.cpython-36.pyc
--- a/magazine/pycache/admin.cpython-36.pyc
+++ b/magazine/pycache/admin.cpython-36.pyc
--- a/magazine/pycache/urls.cpython-36.pyc
+++ b/magazine/pycache/urls.cpython-36.pyc
--- a/magazine/pycache/views.cpython-36.pyc
+++ b/magazine/pycache/views.cpython-36.pyc
--- a/magazine/migrations/pycache/init.cpython-36.pyc
+++ b/magazine/migrations/pycache/init.cpython-36.pyc
--- a/manage.py
+++ b/manage.py
--- a/requirements.txt
+++ b/requirements.txt
@ -0,0 +1,3 @@
+tensorflow
+django
+django-cors-headers
--- a/train_model.py
+++ b/train_model.py
@ -17,12 +17,12 @@ train_x, test_x, train_y, test_y = train_test_split(x, y, test_size=0.2)
 # Building the model
 model = tf.keras.Sequential()

-model.add(layers.Dense(8, activation='relu', input_dim=4))
+model.add(layers.Dense(20, activation='relu', input_dim=4))
 model.add(layers.Dense(3, activation='sigmoid'))

 model.compile(optimizer='adam', loss='categorical_crossentropy',
              metrics=['accuracy'])
 # Training the model
-model.fit(train_x, train_y, validation_data=(test_x, test_y), epochs=2000)
+model.fit(train_x, train_y, validation_data=(test_x, test_y), epochs=1000)

 model.save('iris_model.h5')