This commit is contained in:
Szymon Bartanowicz 2024-05-14 22:58:52 +02:00
parent e8c6c3e7c3
commit d91cded186

View File

@ -35,6 +35,8 @@ pipeline = Pipeline(steps=[
pipeline['model'].compile(optimizer='adam', loss='mse', metrics=['mae']) pipeline['model'].compile(optimizer='adam', loss='mse', metrics=['mae'])
pipeline.fit(X_train.iloc[:, 1:], y_train.iloc[:, 1:], model__epochs=int(sys.argv[1]), model__validation_split=0.1) X_train_excluded = X_train.iloc[:, 1:]
pipeline.fit(X_train_excluded, y_train, model__epochs=10, model__validation_split=0.1)
pipeline['model'].save('powerlifting_model.h5') pipeline['model'].save('powerlifting_model.h5')