fix(recall, precision): little bug in calculations

feat(fmeasure): calculate fmeasure using recall and precision
This commit is contained in:
Mahdi Dibaiee
2016-07-29 22:09:30 +04:30
parent 812717522e
commit 76e7e7faef
2 changed files with 15 additions and 6 deletions

View File

@ -24,6 +24,11 @@ module Main
let showResults (c, r) = putStrLn (classes !! c ++ " ~ " ++ classes !! r)
mapM_ showResults results
let showAccuracy (c, r) =
print $ genericLength (filter (\(h, j) -> h == j && h == c) results) / genericLength results
mapM_ showAccuracy results
putStrLn $ "Recall: " ++ show (recall results) ++ "%"
putStrLn $ "Precision: " ++ show (precision results) ++ "%"
putStrLn $ "F Measure: " ++ show (fmeasure (precision results) (recall results))
putStrLn $ "Accuracy: " ++ show (accuracy results) ++ "%"