Report a bug.

[driftee]

In dataset.py, line 29 and 33, you usepd.read_csv to read from your data that generated from original data. However, you forget to add param header = -1, because the new test.csv and train.csv actully don't have a header. And pandas will use the first line of data as the header, which will cause an index error later.
The follows are what the code should be like.

data = pd.read_csv(os.path.join(root, 'train.txt'), header = -1)
data = pd.read_csv(os.path.join(root, 'test.txt'), header = -1)