diff --git a/function.py b/function.py index b119ab0..783cf28 100644 --- a/function.py +++ b/function.py @@ -87,9 +87,10 @@ def load_data(filepath): data = np.array(data) labels = np.array(labels) - target_variable = labels + data = np.delete(data, [2] + list(range(23, 28)), axis=1) - return data, target_variable + + return data, labels if __name__ == "__main__":