graphstorm
graphstorm copied to clipboard
[Change Request] Modify new `Evaluator` for the three tasks
The current GraphStorm Evaluator
s have misleading naming convention and lack of default metric. The new Evaluator
s will be task-specific.
Sample code
evaluator = gs.eval.GSgnnClassificationEvaluator(eval_frequency=100)
Requested changes
- Create
GSgnnClassificationEvaluator
, setting"accuracy"
as default evaluation metric. - Create
GSgnnRegressionEvaluator
, setting"mse"
as default evaluation metric. - Create
GSgnnLinkPredictionEvaluator
, setting"mrr"
as default evaluation metric.