diff --git a/trainer.py b/trainer.py index 4d80380..94eae35 100644 --- a/trainer.py +++ b/trainer.py @@ -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()