From ea593094ce968b73fe3e8d08de646a8afe397fb3 Mon Sep 17 00:00:00 2001 From: Israel Figueroa Date: Tue, 10 Dec 2024 02:42:30 -0300 Subject: [PATCH] fix --- trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/trainer.py b/trainer.py index 94eae35..c13efb4 100644 --- a/trainer.py +++ b/trainer.py @@ -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()