|
@@ -200,7 +200,7 @@ def main():
|
|
|
model.summary()
|
|
|
|
|
|
print("Fitting model with custom class_weight", class_weight)
|
|
|
- history = model.fit(X_train, y_train, batch_size=16, epochs=3, validation_split = 0.30, verbose=1, shuffle=True, class_weight=class_weight)
|
|
|
+ history = model.fit(X_train, y_train, batch_size=64, epochs=30, validation_split = 0.30, verbose=1, shuffle=True, class_weight=class_weight)
|
|
|
|
|
|
# list all data in history
|
|
|
# print(history.history.keys())
|