[docs]class DataSet:
# raw data (don't touch)
name = None
data = None
[docs] def load_data_set():
raise NotImplementedError
[docs] def make_train_test_split(self, data_set, split_ratio):
raise NotImplementedError
[docs] def get_split(self):
return self.X_train, self.X_test, self.y_train, self.y_test
[docs] def save_data_csv(self, file_name):
self.data.to_csv(file_name)