filipg/geval
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Make Bootstrap work on MultiLabel-F1

Browse Source
This commit is contained in:
Filip Gralinski 2020-01-27 22:14:13 +01:00
parent 43ff22c5dd
commit 659b122625
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/GEval/Core.hs
View File

@ -714,11 +714,14 @@ gevalRunPipeline' parserSpec itemStep finalPipeline context = do
<*> (ZipSource $ recordSource context parserSpec)) .| CL.map (checkStep (Proxy :: Proxy m) itemStep)) .| CL.catMaybes .| finalPipeline)
continueGEvalCalculations :: (MonadIO m) =>
SAMetric t
-> Metric
-> ConduitT (ItemIntermediateRepresentationType t) Void (ResourceT m) MetricOutput
continueGEvalCalculations SAMultiLabelFMeasure (MultiLabelFMeasure beta) = defineContinuation countAgg (fMeasureOnCounts beta) noGraph
continueGEvalCalculations SALikelihood Likelihood = defineContinuation averageC logLossToLikehood noGraph
continueGEvalCalculations SAMultiLabelLikelihood MultiLabelLikelihood = defineContinuation averageC