Commit b4e207e4 authored by Theophile GENTILHOMME's avatar Theophile GENTILHOMME
Browse files

Small changes in titles and legends

parent f2dae89e
Pipeline #19662 passed with stage
in 30 minutes and 30 seconds
......@@ -130,7 +130,7 @@ def lines_at_option(**kwargs):
return list_float_option(
name='lines-at', short_name='la',
desc='If given, draw veritcal lines at the given axis positions',
nitems=None, dflt=None, **kwargs
nitems=None, dflt='1e-3', **kwargs
def x_rotation_option(dflt=0, **kwargs):
......@@ -268,7 +268,8 @@ def criterion_option(lcriteria=['eer', 'hter', 'far'], **kwargs):
ctx.meta['criterion'] = value
return value
return click.option(
'--criterion', default='eer', help='Criterion to compute plots and '
'-c', '--criterion', default='eer',
help='Criterion to compute plots and '
'metrics: `eer` (default), `hter`',
callback=callback, is_eager=True ,**kwargs)(func)
return custom_criterion_option
......@@ -545,10 +545,9 @@ class Hist(PlotBase):
self._criterion = None if 'criterion' not in ctx.meta else \
self._y_label = 'Dev. probability density' if self._eval else \
'density' or self._y_label
self._x_label = 'Scores' if not self._eval else ''
self._title_base = self._title or 'Scores'
self._title_base = 'Scores'
self._y_label = 'Probability Density'
self._x_label = 'Scores values'
self._end_setup_plot = False
def compute(self, idx, input_scores, input_names):
......@@ -565,14 +564,16 @@ class Hist(PlotBase):
self._setup_hist(dev_neg, dev_pos)
mpl.title(self._get_title(idx, dev_file, eval_file))
if not self._eval:
if eval_neg is not None and self._show_dev:
ax = mpl.gca()
#Setup lines, corresponding axis and legends
self._lines(threshold, dev_neg, dev_pos)
if self._eval:
label = "%s threshold" % ('' if self._criterion is None else
self._lines(threshold, label, dev_neg, dev_pos)
if eval_neg is not None:
if self._show_dev:
......@@ -582,10 +583,12 @@ class Hist(PlotBase):
if not self._show_dev:
mpl.title(self._get_title(idx, dev_file, eval_file))
mpl.ylabel('Eval. probability density')
mpl.ylabel('Probability density')
#Setup lines, corresponding axis and legends
self._lines(threshold, eval_neg, eval_pos)
label = "%s threshold (dev)" % ('' if self._criterion is None else
self._lines(threshold, label, eval_neg, eval_pos)
if not self._show_dev:
......@@ -593,13 +596,9 @@ class Hist(PlotBase):
def _get_title(self, idx, dev_file, eval_file):
title = self._legends[idx] if self._legends is not None else None
if title is None:
title = self._title_base if not self._print_fn else \
('%s \n (%s)' % (
str(dev_file) + (" / %s" % str(eval_file) if self._eval else "")
return title
title = title or self._title or self._title_base
title = '' if self._title is not None and not self._title.replace(' ', '') else title
return title or ''
def _plot_legends(self):
lines = []
......@@ -641,8 +640,8 @@ class Hist(PlotBase):
return (n, bins, patches)
def _lines(self, threshold, neg=None, pos=None, **kwargs):
label = 'Threshold' if self._criterion is None else self._criterion.upper()
def _lines(self, threshold, label=None, neg=None, pos=None, **kwargs):
label = label or 'Threshold'
kwargs.setdefault('color', 'C3')
kwargs.setdefault('linestyle', '--')
kwargs.setdefault('label', label)
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment