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

refactor(&&&): use &&& instead of the arrow function

Browse Source
This commit is contained in:
Mahdi Dibaiee 2016-07-29 18:20:35 +04:30
parent b5b4629318
commit 812717522e
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/Sibe/NaiveBayes.hs
View File

@ -16,6 +16,7 @@ module Sibe.NaiveBayes
import qualified Data.Set as Set
import Data.List.Split
import Data.Maybe
import Control.Arrow ((&&&))
type Class = Int
data Document = Document { text :: String
@ -54,7 +55,7 @@ module Sibe.NaiveBayes
createDocuments classes content =
let splitted = splitOn (replicate 10 '-' ++ "\n") content
pairs = map (\a -> ((head . lines) a, (concat . tail . lines) a)) splitted
pairs = map ((head . lines) &&& (concat . tail . lines)) splitted
documents = map (\(topic, text) -> Document text (fromJust $ elemIndex topic classes)) pairs
in documents