jerome-white commited on
Commit
f18e541
1 Parent(s): 7ac39f3

Make CI range more explicit

Browse files
Files changed (1) hide show
  1. app.py +13 -8
app.py CHANGED
@@ -41,7 +41,7 @@ def load(repo):
41
  .query(f'{parameter} == "alpha"')
42
  .drop(columns=parameter))
43
 
44
- def summarize(df, ci=0.95):
45
  def _aggregate(i, g):
46
  values = g['value']
47
  hdi = HDInterval(values)
@@ -115,11 +115,13 @@ class RankPlotter(DataPlotter):
115
  def y(self):
116
  return self.df[self._y]
117
 
118
- def __init__(self, df, top=10):
119
- view = rank(summarize(df), True, self._y)
 
120
  view = (view
121
  .tail(top)
122
  .sort_values(by=self._y, ascending=False))
 
123
  super().__init__(view)
124
 
125
  def draw(self, ax):
@@ -128,7 +130,10 @@ class RankPlotter(DataPlotter):
128
  xmin=self.df['lower'],
129
  xmax=self.df['upper'],
130
  alpha=0.5)
131
- ax.set_xlabel(ax.get_xlabel().title())
 
 
 
132
  ax.set_ylabel('')
133
  ax.set_yticks(self.y, self.df['model'])
134
 
@@ -240,7 +245,7 @@ class DocumentationReader:
240
  #
241
  #
242
  #
243
- def layout(tab):
244
  df = load(Path('jerome-white', tab.dataset))
245
  docs = DocumentationReader(Path('docs', t.docs))
246
 
@@ -249,12 +254,12 @@ def layout(tab):
249
  gr.Markdown(docs['readme'])
250
 
251
  with gr.Column():
252
- plotter = RankPlotter(df)
253
  gr.Plot(plotter.plot())
254
 
255
  with gr.Row():
256
- view = rank(summarize(df), False)
257
- columns = { x.name: f'HDI {x.name}' for x in fields(HDI) }
258
  for i in view.columns:
259
  columns.setdefault(i, i.title())
260
  view = (view
 
41
  .query(f'{parameter} == "alpha"')
42
  .drop(columns=parameter))
43
 
44
+ def summarize(df, ci):
45
  def _aggregate(i, g):
46
  values = g['value']
47
  hdi = HDInterval(values)
 
115
  def y(self):
116
  return self.df[self._y]
117
 
118
+ def __init__(self, df, ci=0.95, top=10):
119
+ self.ci = ci
120
+ view = rank(summarize(df, self.ci), True, self._y)
121
  view = (view
122
  .tail(top)
123
  .sort_values(by=self._y, ascending=False))
124
+
125
  super().__init__(view)
126
 
127
  def draw(self, ax):
 
130
  xmin=self.df['lower'],
131
  xmax=self.df['upper'],
132
  alpha=0.5)
133
+ ax.set_xlabel('{} (with {:.0%} HDI)'.format(
134
+ ax.get_xlabel().title(),
135
+ self.ci,
136
+ ))
137
  ax.set_ylabel('')
138
  ax.set_yticks(self.y, self.df['model'])
139
 
 
245
  #
246
  #
247
  #
248
+ def layout(tab, ci=0.95):
249
  df = load(Path('jerome-white', tab.dataset))
250
  docs = DocumentationReader(Path('docs', t.docs))
251
 
 
254
  gr.Markdown(docs['readme'])
255
 
256
  with gr.Column():
257
+ plotter = RankPlotter(df, ci)
258
  gr.Plot(plotter.plot())
259
 
260
  with gr.Row():
261
+ view = rank(summarize(df, ci), False)
262
+ columns = { x.name: f'{ci:.0%} HDI {x.name}' for x in fields(HDI) }
263
  for i in view.columns:
264
  columns.setdefault(i, i.title())
265
  view = (view