Skip to content

Add int dtype to labels_pred in mutual_info_score()#15462

Merged
TomDLT merged 1 commit intoscikit-learn:masterfrom
ganevgv:docs/add_metrics_cluster_supervised_docs
Nov 2, 2019
Merged

Add int dtype to labels_pred in mutual_info_score()#15462
TomDLT merged 1 commit intoscikit-learn:masterfrom
ganevgv:docs/add_metrics_cluster_supervised_docs

Conversation

@ganevgv
Copy link
Copy Markdown
Contributor

@ganevgv ganevgv commented Nov 2, 2019

Reference Issues/PRs

What does this implement/fix? Explain your changes.

This PR specifies the dtypes of arguments in sklearn.metrics.mutual_info_score, sklearn.metrics.adjusted_mutual_info_score and sklearn.metrics.normalized_mutual_info_score. In the current documentation, the dtype of labels_true is specified to be int but the dtype of labels_pred is not specified. Since our goal is to compare the two clusterings and the metrics are symmetric the two arguments should have the same dtypes.

Any other comments?

@TomDLT TomDLT merged commit b158d0c into scikit-learn:master Nov 2, 2019
@TomDLT
Copy link
Copy Markdown
Member

TomDLT commented Nov 2, 2019

Thanks @ganevgv !

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants