refactor(test): update test script to generate damage files index for dataset_B and adjust export path for processed data

This commit is contained in:
nuluh
2025-04-20 16:02:16 +07:00
parent db2947abdf
commit 1511012e11

View File

@@ -1,12 +1,25 @@
from convert import * from convert import *
from joblib import dump, load from joblib import dump, load
a = generate_damage_files_index( # 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"
# )
b = generate_damage_files_index(
num_damage=6,
file_index_start=1,
col=5,
base_path="D:/thesis/data/dataset_B",
prefix="zzzBD",
) )
data = DataProcessor(file_index=a) # data_A = DataProcessor(file_index=a)
# # data.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.create_vector_column(overwrite=True)
data.create_limited_sensor_vector_column(overwrite=True) data_B.create_limited_sensor_vector_column(overwrite=True)
data.export_to_csv("D:/thesis/data/") data_B.export_to_csv("D:/thesis/data/converted/raw_B")
# a = load("D:/cache.joblib") # a = load("D:/cache.joblib")
breakpoint() # breakpoint()