ajoubrel-ensae/BDC-team-1
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix loading

Browse Source
This commit is contained in:
Alexis REVELLE 2024-03-28 11:39:16 +00:00
parent 840ce876e2
commit d5ab3c2d68
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
0_5_Machine_Learning.py
View File

@ -42,7 +42,7 @@ type_of_model = input('Choisissez le type de model : basique ? premium ?')
S3_ENDPOINT_URL = "https://" + os.environ["AWS_S3_ENDPOINT"]
fs = s3fs.S3FileSystem(client_kwargs={'endpoint_url': S3_ENDPOINT_URL})
dataset_train, dataset_test = load_train_test(type_of_activity)
dataset_train, dataset_test = load_train_test(type_of_activity, type_of_model)
X_train, X_test, y_train, y_test = features_target_split(dataset_train, dataset_test)