nyrasea commited on
Commit
ccb506a
·
verified ·
1 Parent(s): 175a1bf

Update train_model.py

Browse files
Files changed (1) hide show
  1. 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("model/car_model.h5")
57
- print("✅ Model saved to model/car_model.h5")
 
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")