A snippet explaining the usage of confusion_viz

confusion_viz_snippet.py

from confusion_viz import ConfusionViz

y_test = [0, 0, 1, 0, 1]
probas_test = [.1, .2, .5, .7, .9]

conf_viz = ConfusionViz()

conf_viz.fit(
    y_true = y_test, 
    probas_pred = probas_test
)

conf_viz.show()    # shows plot in notebook mode

conf_viz.to_html('confusion_viz.html')    # stores plot as a html file