--model
Browse files
model.py
CHANGED
@@ -29,4 +29,4 @@ model.compile(optimizer='adam', loss='mse')
|
|
29 |
model.fit(X_train_scaled, y_train, epochs=50, batch_size=32, validation_data=(X_test_scaled, y_test))
|
30 |
|
31 |
# Save the trained model
|
32 |
-
model.save('emotion_model')
|
|
|
29 |
model.fit(X_train_scaled, y_train, epochs=50, batch_size=32, validation_data=(X_test_scaled, y_test))
|
30 |
|
31 |
# Save the trained model
|
32 |
+
model.save('models/emotion_model.h5')
|