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

Merge pull request 'fix loading' (#14) from generalization into main

Browse Source 
Reviewed-on: #14
This commit is contained in:
Alexis REVELLE 2024-03-28 12:39:50 +01:00
commit 0aed0911a1
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"] S3_ENDPOINT_URL = "https://" + os.environ["AWS_S3_ENDPOINT"]
fs = s3fs.S3FileSystem(client_kwargs={'endpoint_url': S3_ENDPOINT_URL}) 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) X_train, X_test, y_train, y_test = features_target_split(dataset_train, dataset_test)