DIA
/
mody_2024
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

main
ifiguero 2024-12-10 02:42:30 -03:00
parent fedfe23340
commit ea593094ce
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
trainer.py
View File

@ -657,7 +657,7 @@ class BinaryTuner:
shap.plots.decision(exp.base_values[0], exp.values, features=label_columns, show=False)
plt.title(r"Predicciones mejor modelo: {0}".format(modelname))
plt.ylabel("Predicción del modelo: 0 Negativo, 1 Positivo")
plt.xlabel("Predicción del modelo: 0 Negativo, 1 Positivo")
plt.savefig("{}/shap_{}_{}_{}.png".format(self.name, modelname, dataset, seed),dpi=150, bbox_inches='tight')
plt.close()