[FEAT] Feat Include Undamaged Node Classification #98
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user