gigl.src.training.v1.lib.eval_metrics#

Attributes#

GAUGE_MODEL_METRICS_LOSS

GAUGE_MODEL_METRICS_MRR

HIT_RATE_METRIC_PREFIX

KS_FOR_EVAL

LINK_PREDICTION_METRICS_PREFIX

LOSS_METRIC_NAME

MRR_METRIC_NAME

Functions#

get_metric_name_for_gauge_model_metrics_hit_rate_at(...)

Module Contents#

gigl.src.training.v1.lib.eval_metrics.get_metric_name_for_gauge_model_metrics_hit_rate_at(hit_rate_at)[source]#
Parameters:

hit_rate_at (int)

Return type:

str

gigl.src.training.v1.lib.eval_metrics.GAUGE_MODEL_METRICS_LOSS = 'offline_metric.link_prediction.loss'[source]#
gigl.src.training.v1.lib.eval_metrics.GAUGE_MODEL_METRICS_MRR = 'offline_metric.link_prediction.mrr'[source]#
gigl.src.training.v1.lib.eval_metrics.HIT_RATE_METRIC_PREFIX = 'HitRate_at_'[source]#
gigl.src.training.v1.lib.eval_metrics.KS_FOR_EVAL = [1, 5, 10, 50, 100, 500][source]#
gigl.src.training.v1.lib.eval_metrics.LOSS_METRIC_NAME = 'loss'[source]#
gigl.src.training.v1.lib.eval_metrics.MRR_METRIC_NAME = 'mrr'[source]#