def _get_training_data_x(self):
data_getter = DataGetter()
config.DF_BASE_START_DATE = config.TRAINING_DATE_START
config.DF_BASE_END_DATE = config.TRAINING_DATE_END
X_train = data_getter.get_reshaped_data_for_lstm()
self.X = X_train
config.DF_BASE_START_DATE = config.VALIDATION_DATE_START
config.DF_BASE_END_DATE = config.VALIDATION_DATE_END
X_val = data_getter.get_reshaped_data_for_lstm()
self.X_val = X_val
return X_train, X_val
def _get_testing_data_x(self):
data_getter = DataGetter()
config.DF_BASE_START_DATE = config.TESTING_DATE_START
config.DF_BASE_END_DATE = config.TESTING_DATE_END
reshaped_data_lstm = data_getter.get_reshaped_data_for_lstm()
self.X = reshaped_data_lstm
return reshaped_data_lstm
def test_get_reshaped_data_for_lstm(self):
data_getter = DataGetter()
reshaped_data = data_getter.get_reshaped_data_for_lstm()
self.assertTrue(len(reshaped_data))
utils.write_3d_np_array_to_file("../output/reshaped_data.txt",
reshaped_data)
