Skip to content
Snippets Groups Projects
Commit 0bc5b183 authored by Caron Olivier's avatar Caron Olivier
Browse files

keel dataset loader updated

parent 79d09e1b
No related branches found
No related tags found
No related merge requests found
......@@ -14,7 +14,9 @@ class MyDataSetPreprocessing(DatasetPreprocessing):
v_names, lines_to_avoid = analyse_preamble(dataset_location)
if lines_to_avoid == -1:
raise f"'@data' string not found in CSV file{dataset_location}"
dataset = pd.read_csv(dataset_location, names=v_names, skiprows=lines_to_avoid)
dataset = pd.read_csv(dataset_location, names=v_names,
skiprows=lines_to_avoid, delimiter=',', skipinitialspace=True)
dataset[v_names[-1]] = dataset[v_names[-1]].str.replace(' ', '')
return dataset
except Exception:
raise UnknownDataset(dataset_location)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment