fix(data): Fix pool mapping to include undamaged case and add csv header separator line for Excel compatibility

This commit is contained in:
nuluh
2025-06-18 08:25:01 +07:00
parent 1164627bac
commit a7d8f1ef56
2 changed files with 7 additions and 5 deletions

View File

@@ -25,9 +25,9 @@ def create_ready_data(
"""
ready_data = []
for file in os.listdir(stft_data_path):
ready_data.append(pd.read_csv(os.path.join(stft_data_path, file)))
ready_data.append(pd.read_csv(os.path.join(stft_data_path, file), skiprows=1))
y_data = [i for i in range(len(ready_data))]
y_data = [i for i in range(len(ready_data))] # TODO: Should be replaced with actual desired labels
# Combine all dataframes in ready_data into a single dataframe
if ready_data: # Check if the list is not empty