diff --git a/data/QUGS/test.py b/data/QUGS/test.py index 12b39cf..058af7c 100644 --- a/data/QUGS/test.py +++ b/data/QUGS/test.py @@ -1,25 +1,45 @@ from convert import * from joblib import dump, load +# b = generate_damage_files_index( +# num_damage=6, +# file_index_start=1, +# col=5, +# base_path="D:/thesis/data/dataset_B", +# prefix="zzzBD", +# # undamage_file="zzzBU.TXT" +# ) +# Example: Generate tuples with a special group of df0 at the beginning +special_groups_A = [ + {'df_name': 'zzzAU.TXT', 'position': 0, 'size': 5} # Add at beginning +] + +special_groups_B = [ + {'df_name': 'zzzBU.TXT', 'position': 0, 'size': 5} # Add at beginning +] + +# Generate the tuples with the special group +# a = generate_df_tuples(special_groups=special_groups_A) +b = generate_df_tuples(special_groups=special_groups_B, prefix="zzzBD") + + # a = generate_damage_files_index( -# num_damage=6, file_index_start=1, col=5, base_path="D:/thesis/data/dataset_A" +# num_damage=6, +# file_index_start=1, +# col=5, +# base_path="D:/thesis/data/dataset_A", +# prefix="zzzAD", +# # undamage_file="zzzBU.TXT" # ) -b = generate_damage_files_index( - num_damage=6, - file_index_start=1, - col=5, - base_path="D:/thesis/data/dataset_B", - prefix="zzzBD", -) -# data_A = DataProcessor(file_index=a) -# # data.create_vector_column(overwrite=True) -# data_A.create_limited_sensor_vector_column(overwrite=True) +# data_A = DataProcessor(file_index=a, base_path="D:/thesis/data/dataset_A", include_time=True) +# data_A.create_vector_column(overwrite=True) +# # data_A.create_limited_sensor_vector_column(overwrite=True) # data_A.export_to_csv("D:/thesis/data/converted/raw") -data_B = DataProcessor(file_index=b) -# data.create_vector_column(overwrite=True) -data_B.create_limited_sensor_vector_column(overwrite=True) +data_B = DataProcessor(file_index=b, base_path="D:/thesis/data/dataset_B", include_time=True) +# data_B.create_vector_column(overwrite=True) +# # data_B.create_limited_sensor_vector_column(overwrite=True) data_B.export_to_csv("D:/thesis/data/converted/raw_B") # a = load("D:/cache.joblib") # breakpoint()