Final Version MS2 V2
This commit is contained in:
parent
50b71750ab
commit
f0ebed4731
5 changed files with 2 additions and 2 deletions
|
@ -66,7 +66,7 @@ model.compile(optimizer = 'Adam', loss = 'categorical_crossentropy', metrics = [
|
||||||
history = model.fit(x = X_train_subset,
|
history = model.fit(x = X_train_subset,
|
||||||
y = y_train_subset,
|
y = y_train_subset,
|
||||||
batch_size = 32,
|
batch_size = 32,
|
||||||
epochs = 5,
|
epochs = 100,
|
||||||
verbose = 1,
|
verbose = 1,
|
||||||
validation_data = (X_valid_subset, y_valid_subset))
|
validation_data = (X_valid_subset, y_valid_subset))
|
||||||
|
|
||||||
|
|
|
@ -59,7 +59,7 @@ model.compile(optimizer = 'rmsprop', loss = 'categorical_crossentropy', metrics
|
||||||
history = model.fit(x = X_train_subset,
|
history = model.fit(x = X_train_subset,
|
||||||
y = y_train_subset,
|
y = y_train_subset,
|
||||||
batch_size = 32,
|
batch_size = 32,
|
||||||
epochs = 5,
|
epochs = 500,
|
||||||
verbose = 1,
|
verbose = 1,
|
||||||
validation_data = (X_valid_subset, y_valid_subset))
|
validation_data = (X_valid_subset, y_valid_subset))
|
||||||
|
|
||||||
|
|
Binary file not shown.
Binary file not shown.
Binary file not shown.
Loading…
Reference in a new issue