Spaces:
Runtime error
Runtime error
Update train_model.py
Browse files- train_model.py +2 -2
train_model.py
CHANGED
|
@@ -53,5 +53,5 @@ model.compile(optimizer=Adam(1e-4), loss='categorical_crossentropy', metrics=['a
|
|
| 53 |
model.fit(X_train, y_train, validation_data=(X_test, y_test), epochs=EPOCHS, batch_size=BATCH_SIZE)
|
| 54 |
|
| 55 |
# Simpan model
|
| 56 |
-
model.save("
|
| 57 |
-
print("✅ Model saved to
|
|
|
|
| 53 |
model.fit(X_train, y_train, validation_data=(X_test, y_test), epochs=EPOCHS, batch_size=BATCH_SIZE)
|
| 54 |
|
| 55 |
# Simpan model
|
| 56 |
+
model.save("car_model.h5")
|
| 57 |
+
print("✅ Model saved to car_model.h5")
|