From dd5c3f416b2c140d6d1b557f614d299ed1f45e9f Mon Sep 17 00:00:00 2001 From: ajoubrel-ensae Date: Tue, 26 Mar 2024 21:07:29 +0000 Subject: [PATCH] v2 --- 0_3_General_modelization_dataset.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/0_3_General_modelization_dataset.py b/0_3_General_modelization_dataset.py index 2feb2a0..a79ea72 100644 --- a/0_3_General_modelization_dataset.py +++ b/0_3_General_modelization_dataset.py @@ -21,7 +21,7 @@ warnings.filterwarnings('ignore') # functions def generate_test_set(type_of_comp): - file_path_list = fs.ls(f"projet-bdc2324-team1/Generalization/{type_of_comp}/Test_set") + file_path_list = fs.ls(f"projet-bdc2324-team1/Generalization_v2/{type_of_comp}/Test_set") test_set = pd.DataFrame() for file in file_path_list: print(file) @@ -32,7 +32,7 @@ def generate_test_set(type_of_comp): def generate_train_set(type_of_comp): - file_path_list = fs.ls(f"projet-bdc2324-team1/Generalization/{type_of_comp}/Train_set") + file_path_list = fs.ls(f"projet-bdc2324-team1/Generalization_v2/{type_of_comp}/Train_set") train_set = pd.DataFrame() for file in file_path_list: print(file) @@ -43,7 +43,7 @@ def generate_train_set(type_of_comp): type_of_comp = input('Choisissez le type de compagnie : sport ? musique ? musee ?') -BUCKET_OUT = f'projet-bdc2324-team1/Generalization/{type_of_comp}/' +BUCKET_OUT = f'projet-bdc2324-team1/Generalization_v2/{type_of_comp}/' # create test and train datasets test_set = generate_test_set(type_of_comp)