From 824c2f041c67ef56951522d9b5198f5ed15efe51 Mon Sep 17 00:00:00 2001 From: Tanushree Tunstall Date: Fri, 18 Mar 2022 17:50:24 +0000 Subject: [PATCH] saving work --- hyperparams.py | 4 ---- imports.py | 7 ++++++- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/hyperparams.py b/hyperparams.py index b877fa9..e51324d 100644 --- a/hyperparams.py +++ b/hyperparams.py @@ -91,10 +91,6 @@ gs_gb = GridSearchCV(estimator=XGBClassifier(n_jobs=-1), n_jobs=4, refit='recall') #%% - - - -#%% gs_mlp = MLPClassifier(max_iter=600) clf = GridSearchCV(gs_mlp, parameter_space, n_jobs= 10, cv = 10) diff --git a/imports.py b/imports.py index 62ba294..b5938a3 100644 --- a/imports.py +++ b/imports.py @@ -9,7 +9,7 @@ import os, sys import pandas as pd import numpy as np import pprint as pp -#from copy import deepcopy +from copy import deepcopy from sklearn import linear_model from sklearn.linear_model import LogisticRegression, LinearRegression from sklearn.naive_bayes import BernoulliNB @@ -30,6 +30,8 @@ from sklearn.compose import make_column_transformer from sklearn.metrics import confusion_matrix, accuracy_score, precision_score, recall_score from sklearn.metrics import roc_auc_score, roc_curve, f1_score, matthews_corrcoef +from sklearn.metrics import jaccard_score + from sklearn.metrics import make_scorer from sklearn.metrics import classification_report @@ -71,6 +73,7 @@ scoring_fn = ({'accuracy' : make_scorer(accuracy_score) , 'precision' : make_scorer(precision_score) , 'recall' : make_scorer(recall_score) , 'roc_auc' : make_scorer(roc_auc_score) + , 'jcc' : make_scorer(jaccard_score) }) rs = {'random_state': 42} @@ -78,6 +81,8 @@ njobs = {'n_jobs': 10} skf_cv = StratifiedKFold(n_splits = 10 #, shuffle = False, random_state= None) , shuffle = True,**rs) +#my_mcc = make_scorer({'mcc':make_scorer(matthews_corrcoef}) +mcc_score_fn = {'mcc': make_scorer(matthews_corrcoef)} #%% homedir = os.path.expanduser("~")