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

Fix bug with no calibration plot for Probabilistic-MultiLabel-F

Browse Source
This commit is contained in:
Filip Gralinski 2021-07-23 17:38:57 +02:00
parent c71bba81f3
commit 36a930d159
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/GEval/OptionsParser.hs
View File

@ -471,7 +471,13 @@ groupByMetric metrics results = filter (\(_, ss) -> not (null ss))
plotGraph :: FilePath -> (EvaluationScheme, [(SourceSpec, GraphSeries)]) -> IO () plotGraph :: FilePath -> (EvaluationScheme, [(SourceSpec, GraphSeries)]) -> IO ()
plotGraph graphFile (scheme@(EvaluationScheme (ProbabilisticSoftFMeasure _) _), seriesSpecs) = do plotGraph graphFile (scheme@(EvaluationScheme (ProbabilisticSoftFMeasure _) _), seriesSpecs)
= calibrationPlot graphFile scheme seriesSpecs
plotGraph graphFile (scheme@(EvaluationScheme (ProbabilisticMultiLabelFMeasure _) _), seriesSpecs)
= calibrationPlot graphFile scheme seriesSpecs
plotGraph _ (scheme, _) = error $ "No graph for metric!" ++ (show scheme)
calibrationPlot graphFile scheme seriesSpecs = do
toFile def graphFile $ do toFile def graphFile $ do
layoutlr_title .= "GEval Graph / Calibration / Loess / " ++ (show scheme) layoutlr_title .= "GEval Graph / Calibration / Loess / " ++ (show scheme)
let perfectSeries = (FilePathSpec "Perfect", let perfectSeries = (FilePathSpec "Perfect",
@ -481,7 +487,6 @@ plotGraph graphFile (scheme@(EvaluationScheme (ProbabilisticSoftFMeasure _) _),
where where
plotOneSeries :: (SourceSpec, GraphSeries) -> EC (LayoutLR Double Double Double) () plotOneSeries :: (SourceSpec, GraphSeries) -> EC (LayoutLR Double Double Double) ()
plotOneSeries (sspec, GraphSeries series) = plotLeft (line (recoverPath sspec) [series]) plotOneSeries (sspec, GraphSeries series) = plotLeft (line (recoverPath sspec) [series])
plotGraph _ _ = error "No graph for this metric!"
initChallenge :: GEvalSpecification -> IO () initChallenge :: GEvalSpecification -> IO ()