This commit is contained in:
parent
3d068ac7d1
commit
07e0eb04eb
4
train.py
4
train.py
@ -43,10 +43,8 @@ def prepare_model(epoch_param, batch_param, _run):
|
|||||||
X_test = vgsales_test.drop(['Rank','Name','Platform','Year','Genre','Publisher'],axis = 1)
|
X_test = vgsales_test.drop(['Rank','Name','Platform','Year','Genre','Publisher'],axis = 1)
|
||||||
y_test = vgsales_test[['Nintendo']]
|
y_test = vgsales_test[['Nintendo']]
|
||||||
|
|
||||||
print(X_train.shape[1])
|
|
||||||
|
|
||||||
model = Sequential()
|
model = Sequential()
|
||||||
model.add(Dense(9, input_dim = X_train.shape[1], kernel_initializer='normal', activation='relu'))
|
model.add(Dense(9, input_dim = X_train, kernel_initializer='normal', activation='relu'))
|
||||||
model.add(Dense(1,kernel_initializer='normal', activation='sigmoid'))
|
model.add(Dense(1,kernel_initializer='normal', activation='sigmoid'))
|
||||||
|
|
||||||
early_stop = EarlyStopping(monitor="val_loss", mode="min", verbose=1, patience=10)
|
early_stop = EarlyStopping(monitor="val_loss", mode="min", verbose=1, patience=10)
|
||||||
|
Loading…
Reference in New Issue
Block a user