SocialED.metrics

metrics

Evaluation metrics for clustering and classification in social event detection.

SocialED.metrics.metric.eval_nmi(ground_truths, predictions)[source]

Normalized Mutual Information (NMI) score for clustering evaluation.

Parameters:
  • ground_truths (array-like) – Ground truth labels.

  • predictions (array-like) – Predicted cluster labels.

Returns:

nmi – Normalized Mutual Information score.

Return type:

float

SocialED.metrics.metric.eval_ami(ground_truths, predictions)[source]

Adjusted Mutual Information (AMI) score for clustering evaluation.

Parameters:
  • ground_truths (array-like) – Ground truth labels.

  • predictions (array-like) – Predicted cluster labels.

Returns:

ami – Adjusted Mutual Information score.

Return type:

float

SocialED.metrics.metric.eval_ari(ground_truths, predictions)[source]

Adjusted Rand Index (ARI) score for clustering evaluation.

Parameters:
  • ground_truths (array-like) – Ground truth labels.

  • predictions (array-like) – Predicted cluster labels.

Returns:

ari – Adjusted Rand Index score.

Return type:

float

SocialED.metrics.metric.eval_f1(ground_truths, predictions)[source]

F1 score for classification evaluation.

Parameters:
  • ground_truths (array-like) – Ground truth labels.

  • predictions (array-like) – Predicted labels.

Returns:

f1 – F1 score.

Return type:

float

SocialED.metrics.metric.eval_acc(ground_truths, predictions)[source]

Accuracy score for classification evaluation.

Parameters:
  • ground_truths (array-like) – Ground truth labels.

  • predictions (array-like) – Predicted labels.

Returns:

acc – Accuracy score.

Return type:

float