diff --git a/geval.cabal b/geval.cabal index e9acd33..8e7827f 100644 --- a/geval.cabal +++ b/geval.cabal @@ -1,5 +1,5 @@ name: geval -version: 0.2.0.0 +version: 0.2.1.0 synopsis: Machine learning evaluation tools description: Please see README.md homepage: http://github.com/name/project diff --git a/src/GEval/Core.hs b/src/GEval/Core.hs index 527d15e..fabeb10 100644 --- a/src/GEval/Core.hs +++ b/src/GEval/Core.hs @@ -40,6 +40,14 @@ type MetricValue = Double data Metric = RMSE | MSE | BLEU | Accuracy deriving (Show, Read) +data MetricOrdering = TheLowerTheBetter | TheHigherTheBetter + +getMetricOrdering :: Metric -> MetricOrdering +getMetricOrdering RMSE = TheLowerTheBetter +getMetricOrdering MSE = TheLowerTheBetter +getMetricOrdering BLEU = TheHigherTheBetter +getMetricOrdering Accuracy = TheHigherTheBetter + defaultOutDirectory = "." defaultTestName = "test-A" defaultOutFile = "out.tsv"