def __init__(self,
scores,
new_golds=True,
thresholds=None,
gold_getter=None):
"""
Pararmeters
-----------
scores : {Score}
Mapping of scores in export
new_golds : bool
Flag to indicate whether to fetch gold labels from database
or to use the gold labels already in score objects
"""
if gold_getter is None:
gold_getter = GoldGetter()
gold_getter.all()
self.gold_getter = gold_getter
if new_golds:
scores = self._init_golds(scores)
self.scores = scores
self._sorted_ids = sorted(scores, key=lambda id_: scores[id_].p)
self.class_counts = ScoreStats.counts(self.sorted_scores)
if thresholds is None:
thresholds = self.find_thresholds(config.fpr, config.mdr)
self.thresholds = thresholds
self._stats = None
def test_wrapper_getter(self):
gg = GoldGetter()
gg._golds = {}
gg.all()
print(gg.getters)
assert callable(gg.getters[0])
def __init__(self, history, gold_getter=None):
"""
Parameters
----------
history : {History}
Mapping of Subject History to subject id
"""
self.history = history
if gold_getter is None:
gold_getter = GoldGetter()
gold_getter.all()
self.gold_getter = gold_getter
def _run(self):
gg = GoldGetter()
gg.all()
swap = self.init_swap()
gi = GoldIterator(gg.golds, self.start, self.step)
for n, golds in enumerate(gi):
logger.info('Running trial %d with %d golds', n, len(golds))
fake = 0
for gold in golds.values():
if gold == -1:
fake += 1
logger.debug('Fake n golds: %d', fake)
swap.set_gold_labels(golds)
swap.process_changes()
self.add_trial(randomex.Trial(n, golds, swap.score_export()))
if len(golds) > self.end:
break
def test_wrapper_golds_to_None(self):
gg = GoldGetter()
gg._golds = {}
gg.all()
assert gg._golds is None