s434695/ium_434695
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix

Browse Source
This commit is contained in:
s434695 2021-05-23 23:25:50 +02:00
parent 30c8343c6c
commit 069a5860d6
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -40,10 +40,8 @@ def prepare_model(epoch_param, batch_param, _run):
X_train = vgsales_train.drop(['Rank','Name','Platform','Year','Genre','Publisher'],axis = 1)
y_train = vgsales_train[['Nintendo']]
y_train = keras.utils.to_categorical(y_train)
X_test = vgsales_test.drop(['Rank','Name','Platform','Year','Genre','Publisher'],axis = 1)
y_test = vgsales_test[['Nintendo']]
y_test = keras.utils.to_categorical(y_test)
print(X_train.shape[1])
@ -53,7 +51,7 @@ def prepare_model(epoch_param, batch_param, _run):
early_stop = EarlyStopping(monitor="val_loss", mode="min", verbose=1, patience=10)
model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
epochs = int(sys.argv[1])
batch_size = int(sys.argv[2])