diff --git a/src/GEval/Core.hs b/src/GEval/Core.hs index 1fde07d..0ddc736 100644 --- a/src/GEval/Core.hs +++ b/src/GEval/Core.hs @@ -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