gli
gli copied to clipboard
Add node regression in benchmarking code
As titled.
Provide mse and mae as metrics and loss function.
train.py can run successfully on cora (pretending it is a node regression task).
TODO: update train_minibatch.py and train_gbdt.py
Update:
-
remove redundant codes
-
fix
train_minibatch.py, training is successful
-
fix typo in
train.py