various bugs

This commit is contained in:
Tanushree Tunstall 2022-07-10 20:00:35 +01:00
parent 6950c4b057
commit 33e3b5a0a6
3 changed files with 39 additions and 24 deletions

View file

@ -354,9 +354,10 @@ def MultModelsCl(input_df, target
y_pred = cross_val_predict(model_pipeline
, input_df
, target
, cv = sel_cv
#, groups = group
, **njobs)
#, commented out thing,
, cv=sel_cv
, **njobs
)
#_tn, _fp, _fn, _tp = confusion_matrix(y_pred, y).ravel() # internally
tn, fp, fn, tp = confusion_matrix(y_pred, target).ravel()

View file

@ -48,8 +48,13 @@ ml_gene_drugD = {'pncA' : 'pyrazinamide'
, 'gid' : 'streptomycin'
}
gene_dataD={}
split_types = ['70_30', '80_20', 'sl']
split_data_types = ['actual', 'complete']
split_types = ['70_30',
'80_20',
'sl'
]
split_data_types = ['actual',
'complete'
]
for gene, drug in ml_gene_drugD.items():
print ('\nGene:', gene