def get_data_all(path_data, array_max): data_all = Dataset_csv(path_data=path_data, max_value=array_max) data_all.set_minibatch(data_all.total_inputs) X_data, y_data = data_all.generate_batch() return X_data, y_data, len(y_data)
def get_data_split(path_data, array_max, test_size=0.3): data_all = Dataset_csv(path_data=path_data, max_value=array_max) data_all.set_minibatch(data_all.total_inputs) data, label = data_all.generate_batch() X_train, X_test, y_train, y_test = model_selection.train_test_split( data, label, test_size=test_size, random_state=42) return X_train, X_test, y_train, y_test, len(y_train), len(y_test)
def get_data_all(path_train, path_test, array_max): data_all = Dataset_csv(path_data=path_train, max_value=array_max) data_all.set_minibatch(data_all.total_inputs) X_train, y_train = data_all.generate_batch() data_all = Dataset_csv(path_data=path_test, max_value=array_max) data_all.set_minibatch(data_all.total_inputs) X_test, y_test = data_all.generate_batch() return X_train, X_test, y_train, y_test, len(y_train), len(y_test)