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

fix loading #14

Merged
arevelle-ensae merged 1 commits from generalization into main 2024-03-28 12:39:51 +01:00
Conversation 0 Commits 1 Files Changed 1 +1 -1
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)