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

v2

Browse Source
This commit is contained in:
Antoine JOUBREL 2024-03-26 21:07:29 +00:00
parent 5e37dd4d3d
commit dd5c3f416b
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
0_3_General_modelization_dataset.py
View File

@ -21,7 +21,7 @@ warnings.filterwarnings('ignore')
# functions # functions
def generate_test_set(type_of_comp): 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() test_set = pd.DataFrame()
for file in file_path_list: for file in file_path_list:
print(file) print(file)
@ -32,7 +32,7 @@ def generate_test_set(type_of_comp):
def generate_train_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() train_set = pd.DataFrame()
for file in file_path_list: for file in file_path_list:
print(file) 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 ?') 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 # create test and train datasets
test_set = generate_test_set(type_of_comp) test_set = generate_test_set(type_of_comp)