epitome.viz.joint_plot

epitome.viz.joint_plot(dict_model1, dict_model2, metric='AUC', model1_name='model1', model2_name='model2', outlier_filter=None)

Returns seaborn joint plot of two models.

Parameters

  • dict_model1 (dict) – dictionary of TF: metrics for first model. Output from run_predictions().

  • dict_model2 (doct) – dictionary of TF: metrics for second model. Output from run_predictions().

  • metric (dict) – metric in dicts. Should be auPRC, AUC, or GINI.

  • model1_name (str) – string for model 1 name, shown on x axis.

  • model2_name (str) – string for model 2 name, shown on y axis.

  • outlier_filter (str) – string filter to label. Defaults to no labels.

Returns

matplotlib axis

Return type

matplotlib axis