raise SystemExit("params folder exists! select a new params path please") class MyInit(mx.init.Initializer): xavier = mx.init.Xavier() uniform = mx.init.Uniform() def _init_weight(self, name, data): if len(data.shape) < 2: self.uniform._init_weight(name, data) print('Init', name, data.shape, 'with Uniform') else: self.xavier._init_weight(name, data) print('Init', name, data.shape, 'with Xavier') if __name__ == "__main__": # read all data from graph singal matrix file all_data = read_and_generate_dataset(graph_signal_matrix_filename, num_of_vertices, num_of_features, num_of_weeks, num_of_days, num_of_hours, points_per_hour, num_for_predict) # test set ground truth true_value = all_data['test']['target'].transpose((0, 2, 1)).reshape(all_data['test']['target'].shape[0], -1) # training set data loader train_loader = gluon.data.DataLoader( gluon.data.ArrayDataset( nd.array(all_data['train']['week'], ctx = ctx), nd.array(all_data['train']['day'], ctx = ctx), nd.array(all_data['train']['recent'], ctx = ctx), nd.array(all_data['train']['target'], ctx = ctx) ), batch_size = batch_size, shuffle = True )
# check parameters file if os.path.exists(params_path) and not FLAGS.force: raise SystemExit("Params folder exists! Select a new params path please!") else: if os.path.exists(params_path): shutil.rmtree(params_path) os.makedirs(params_path) print('Create params directory %s' % (params_path)) if __name__ == "__main__": # read all data from graph signal matrix file print("Reading data...") #Input: train / valid / test : length x 3 x NUM_POINT x 12 all_data = read_and_generate_dataset(graph_signal_matrix_filename, num_of_weeks, num_of_days, num_of_hours, num_for_predict, points_per_hour, merge) # test set ground truth true_value = all_data['test']['target'] print(true_value.shape) # training set data loader train_loader = DataLoader(TensorDataset( torch.Tensor(all_data['train']['week']), torch.Tensor(all_data['train']['day']), torch.Tensor(all_data['train']['recent']), torch.Tensor(all_data['train']['target'])), batch_size=batch_size, shuffle=True)