fix data type
All checks were successful
s464980-training/pipeline/head This commit looks good

This commit is contained in:
Sheaza 2024-05-14 21:44:07 +02:00
parent 34ff694fcd
commit 236863587a

View File

@ -48,6 +48,6 @@ parser.add_argument('--epochs')
args = parser.parse_args()
model = RegressionModel()
model.load_data("df_train.csv", "df_test.csv")
model.train(epochs=args.epochs)
model.train(epochs=int(args.epochs))
model.evaluate()
model.save_model()