thereadme/sibe
1
0
Fork 0
Code Issues 1 Pull Requests Projects Releases Wiki Activity

feat(results): accuracy, recall and precision functions used to calculate measures

Browse Source 
fix: read data from another repository
This commit is contained in:
Mahdi Dibaiee
2016-07-29 17:55:59 +04:30
parent eeabe4696c
commit b5b4629318
6 changed files with 49 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/doc-classifier-data Submodule

Submodule examples/doc-classifier-data added at 5b069a54a6

20
examples/naivebayes-doc-classifier.hs
View File

@ -6,24 +6,24 @@ module Main
import Data.List
import Data.Maybe
import Debug.Trace
import Data.List.Split
main = do
dataset <- readFile "examples/naivebayes-doc-classifier/data-reuters"
test <- readFile "examples/naivebayes-doc-classifier/data-reuters-test"
dataset <- readFile "examples/doc-classifier-data/data-reuters"
test <- readFile "examples/doc-classifier-data/data-reuters-test"
classes <- map (filter (/= ' ')) . lines <$> readFile "examples/naivebayes-doc-classifier/data-classes"
classes <- map (filter (/= ' ')) . lines <$> readFile "examples/doc-classifier-data/data-classes"
let intClasses = [0..length classes - 1]
documents = createDocuments classes dataset
testDocuments = createDocuments classes test
nb = initialize documents
let testResults (Document text c) =
let r = determine text nb intClasses documents
in trace (classes !! c ++ " ~ " ++ classes !! r) c == r
results = map (\(Document text c) -> (c, determine text nb intClasses documents)) testDocuments
let results = map testResults testDocuments
let showResults (c, r) = putStrLn (classes !! c ++ " ~ " ++ classes !! r)
mapM_ showResults results
putStr "Accuracy: "
putStr . show . round $ (genericLength (filter (==True) results) / genericLength results) * 100
putStrLn "%"
putStrLn $ "Recall: " ++ show (recall results) ++ "%"
putStrLn $ "Precision: " ++ show (precision results) ++ "%"
putStrLn $ "Accuracy: " ++ show (accuracy results) ++ "%"