changed script
This commit is contained in:
parent
ba6c8d5dde
commit
1e044d743f
2
model.py
2
model.py
@ -35,6 +35,6 @@ pipeline = Pipeline(steps=[
|
||||
|
||||
pipeline['model'].compile(optimizer='adam', loss='mse', metrics=['mae'])
|
||||
|
||||
pipeline.fit(X_train, y_train, model__epochs=int(sys.argv[1]), model__validation_split=0.1)
|
||||
pipeline.fit(X_train.iloc[:, 1:], y_train, model__epochs=int(sys.argv[1]), model__validation_split=0.1)
|
||||
|
||||
pipeline['model'].save('powerlifting_model.h5')
|
||||
|
Loading…
Reference in New Issue
Block a user