def load_data(net_type, num_step):
dataReader = PM25DataReader(net_type, num_step)
dataReader.ReadData()
dataReader.Normalize()
dataReader.GenerateValidationSet(k=1000)
x_train, y_train = dataReader.XTrain, dataReader.YTrain
x_test, y_test = dataReader.XTest, dataReader.YTest
x_val, y_val = dataReader.XDev, dataReader.YDev
return x_train, y_train, x_test, y_test, x_val, y_val
def load_data(net_type, num_step):
dr = PM25DataReader(net_type, num_step)
dr.ReadData()
dr.Normalize()
dr.GenerateValidationSet(k=1000)
return dr