cluster_v_measure#

er_evaluation.cluster_v_measure(prediction, reference, beta=1.0)[source]#

Compute the V-measure.

This wraps scikit-learn’s V-measure function.

Parameters:

  • prediction (Series) – membership vector for predicted clusters, i.e. a pandas Series indexed by mention ids and with values representing predicted cluster assignment.

  • reference (Series) – membership vector for reference (true) clusters, i.e. a pandas Series indexed by mention ids and with values representing reference cluster assignment.

Returns:

V-measure

Return type:

float

Notes

  • The prediction and reference membership vectors are inner joined before this metric is computed.

  • NA values are dropped from membership vectors prior to computing the metric.