Removed changes for shorter run

2025-09-04 13:45:34 +02:00
parent b1b63d416d
commit 4adb507d6a
2 changed files with 2 additions and 3 deletions
@@ -135,7 +135,6 @@ def main_two():
                    file_path = os.path.join(dataset_path, data_filename)
                    df = load_dataset(file_path)
                    df = remove_covid_data(df)
-                    df = df.head(1000) # TODO: remove
                    tr,val,te = split_method(df, percentages=(80,10,10))
                    tr = reduce_columns(tr, data_filename)
                    val = reduce_columns(val, data_filename)