Skip to content
This repository has been archived by the owner on Mar 22, 2024. It is now read-only.

fix classification tasks #25

Merged
merged 4 commits into from
Dec 23, 2020
Merged

fix classification tasks #25

merged 4 commits into from
Dec 23, 2020

Conversation

manonreau
Copy link
Collaborator

No description provided.

Copy link
Member

@NicoRenaud NicoRenaud left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Very minor comment otherwise looks good !

@@ -6,46 +6,46 @@
from sklearn.metrics import confusion_matrix


def get_boolean(vals, threshold, target):
def get_boolean(values, threshold, target):

if target == 'fnat' or target == 'bin':
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

you could change that to:

inverse = ['fnat', 'bin']
if target in inverse:
    ..... 

so that if we add more targets like fnat we can simply add them to the list

@manonreau manonreau merged commit 486baa1 into master Dec 23, 2020
@manonreau manonreau deleted the class branch January 18, 2021 15:06
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants