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

commit

main
ifiguero 2024-12-10 01:16:23 -03:00
parent 44d5456d0a
commit fedfe23340
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
trainer.py
View File

@ -656,8 +656,8 @@ class BinaryTuner:
feature_names=label_columns)
shap.plots.decision(exp.base_values[0], exp.values, features=label_columns, show=False)
plt.title(r"{0}".format(modelname))
plt.ylabel("Respuesta del Modelo: 0 Negativo, 1 Positivo")
plt.title(r"Predicciones mejor modelo: {0}".format(modelname))
plt.ylabel("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()
@ -673,8 +673,8 @@ class BinaryTuner:
feature_names=label_columns)
shap.plots.decision(exp.base_values[0], exp.values, features=label_columns, show=False)
plt.title(r"{0}".format(modelname))
plt.ylabel("Respuesta del Modelo Positivas")
plt.title(r"Predicciones mejor modelo: {0}=1".format(modelname))
plt.xlabel("Predicción del modelo")
plt.savefig("{}/shap_pos_{}_{}_{}.png".format(self.name, modelname, dataset, seed),dpi=150, bbox_inches='tight')
plt.close()
@ -686,9 +686,9 @@ class BinaryTuner:
feature_names=label_columns)
shap.plots.decision(exp.base_values[0], exp.values, features=label_columns, show=False)
plt.title(r"{0}".format(modelname))
plt.ylabel("Respuesta del Modelo Negativas")
plt.savefig("{}/shap_pos_{}_{}_{}.png".format(self.name, modelname, dataset, seed),dpi=150, bbox_inches='tight')
plt.title(r"Predicciones mejor modelo: {0}=0".format(modelname))
plt.xlabel("Predicción del modelo")
plt.savefig("{}/shap_neg_{}_{}_{}.png".format(self.name, modelname, dataset, seed),dpi=150, bbox_inches='tight')
plt.close()