Skip to content
Snippets Groups Projects

Add a predictions algorithm useful for deeplearning

Merged Amir MOHAMMADI requested to merge predictions into master
4 files
+ 44
6
Compare changes
  • Side-by-side
  • Inline
Files
4
+ 14
0
from bob.pad.base.algorithm import Algorithm
class Predictions(Algorithm):
"""An algorithm that takes the precomputed predictions and uses them for
scoring."""
def __init__(self, **kwargs):
super(Predictions, self).__init__(
**kwargs)
def score(self, predictions):
# Assuming the predictions are the output of a softmax layer
return [predictions[1]]
Loading