s464980/IUM_s464980
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Sheaza 2024-05-14 21:44:07 +02:00
parent 34ff694fcd
commit 236863587a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

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