Hyperparameters for other datasets

[kevinzhangcode]

• Q1: About the training parameters of other datasets

• I noticed that only the hyperparameters of some datasets are released in the yaml file. Could you please release the training parameters of other datasets?

• Q2: I'm trying to run this work in colab, but find F1-score can't reach 79.12.

config['dataset']='WikiCS'
config

{'seed': 12345,
 'learning_rate': 0.0001,
 'num_hidden': 256,
 'num_input_dim': 1433,
 'num_proj_hidden': 256,
 'num_layers': 2,
 'num_proj_layers': 2,
 'drop_edge_rate_1': 0.2,
 'drop_edge_rate_2': 0.4,
 'drop_feature_rate_1': 0.3,
 'drop_feature_rate_2': 0.4,
 'tau': 0.4,
 'num_epochs': 700,
 'eta': 0.1,
 'K': 1433,
 'm': 1,
 'gpu_id': 0,
 'root': '/content/drive/MyDrive/COSTA-main',
 'model': 'COSTA',
 'dataset': 'WikiCS',
 'data_dir': '/content/drive/MyDrive/COSTA-main/data',
 'test_every_epoch': False}

Runner(conf=config).execute()

/usr/local/lib/python3.7/dist-packages/torch_geometric/datasets/wikics.py:39: UserWarning: The WikiCS dataset now returns an undirected graph by default. Please explicitly specify 'is_undirected=False' to restore the old behaviour.
  f"The {self.__class__.__name__} dataset now returns an "
Initializing and building model...
Load Dataset...
Training Model...
(T): 100%|██████████| 700/700 [07:49<00:00,  1.49it/s, loss=6.93]
Testing...
(LR): 100%|██████████| 5000/5000 [00:07<00:00, best test F1Mi=0.278, F1Ma=0.0923](E): Best test F1Mi=0.2784, F1Ma=0.0923

[ShenLongDX]

我遇到了相同的情况，对于wikics的结果无法复现