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

fix

main
ifiguero 2024-12-10 02:44:47 -03:00
parent ea593094ce
commit 85571895de
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
trainer.py
View File

@ -700,13 +700,13 @@ class BinaryTuner:
for i in range(5):
shap.plots.waterfall(exp[i], show=False)
plt.title(r"{0} $y_{{{1}}}=0$".format(modelname, i))
plt.title(r"{0} $y_{{{1}}}=1$".format(modelname, i))
plt.savefig("{}/pos_{}_{}_{}_{}.png".format(self.name, i, modelname, dataset, seed),dpi=150, bbox_inches='tight')
plt.close()
for i in range(5, 10):
shap.plots.waterfall(exp[i], show=False)
plt.title(r"{0} $y_{{{1}}}=1$".format(modelname, i-5))
plt.title(r"{0} $y_{{{1}}}=0$".format(modelname, i-5))
plt.savefig("{}/neg_{}_{}_{}_{}.png".format(self.name, i-5, modelname, dataset, seed),dpi=150, bbox_inches='tight')
plt.close()