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)