fix data type

2024-05-14 21:44:07 +02:00 · 2024-05-14 21:44:07 +02:00 · 236863587a
parent 34ff694fcd
commit 236863587a
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -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()