Test.hs

{-# LANGUAGE OverloadedStrings            #-}

module Main where

import Data.FastText
import System.Exit          (ExitCode (..), exitWith)

main :: IO ()
main = do
  model <- loadModel "data/lid.176.ftz"
  putStrLn $ "Loaded model with dimension " <> show (getDimension model)
  let input = "og overalt gaar det tidligere undertrykte maal stadig fremad, trods brølene og trods dannelsen — akkurat som her"
      expLabel = "__label__no"
  res <- predictProbs model 2 0 input
  case res of
    (Prediction _ l) : _ | l == expLabel -> print res >> exitWith ExitSuccess
    _ -> do
      putStrLn $ "Test failed, expected " <> show expLabel <> " as top label, result, got:\n " <> show res <> " ... "
      exitWith (ExitFailure 1)