Script to compute predictability of smart paradigms

1 files changed, 120 insertions, 0 deletions

diff --git a/scripts/rgl-predictability.hs b/scripts/rgl-predictability.hs
new file mode 100644
index 000000000..ec1abefed
--- /dev/null
+++ b/scripts/rgl-predictability.hs
@@ -0,0 +1,120 @@
+{-# LANGUAGE OverloadedStrings #-}
+module Main where
+
+import GF.Predictability
+
+main = defaultMain
+  [ englishNouns, englishVerbs
+  -- , finnishNouns, finnishVerbs ]
+  , frenchVerbs
+  , swedishNouns, swedishVerbs ]
+
+englishNouns, englishVerbs :: Experiment
+englishNouns = Experiment
+  { title = "English nouns"
+  , lexicon = "lib/src/english/DictEng.gf"
+  , category = "N"
+  , nforms        = 4
+  , morphology = "lib/src/english/ParadigmsEng.gf"
+  , smartparadigm = "mkN"
+  , setup = \(man:_:men:_:_) -> [ [esc man], [esc man,esc men] ] }
+englishVerbs = Experiment
+  { title = "English verbs"
+  , lexicon = "lib/src/english/DictEng.gf"
+  , category = "V"
+  , nforms        = 5
+  , morphology = "lib/src/english/ParadigmsEng.gf"
+  , smartparadigm = "mkV"
+  , setup = \(blow:blows:blown:blowing:blew:_) ->
+    [ [ esc blow ]
+    , [ esc blow, esc blew ]
+    , [ esc blow, esc blew, esc blown ]
+    , [ esc blow, esc blows, esc blew, esc blown, esc blowing ] ] }
+
+  -- ~~~ Finnish ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+finnishNouns, finnishVerbs :: Experiment
+finnishNouns = Experiment
+  { title = "Finnish nouns"
+  , lexicon = "lib/src/finnish/DictFin.gf"
+  , category = "NK"
+  , nforms        = 10
+  , morphology = "lib/src/finnish/ParadigmsFin.gf"
+  , smartparadigm = "mkN"
+  , setup = \(talo:talon:taloa:talona:taloon:talojen:taloja:taloina:taloissa:taloihin:_) ->
+      map (map esc )
+        [ [ talo ]
+        , [ talo, taloja ]
+        , [ talo, talon, taloja ]
+        , [ talo, talon, taloja, taloa ]
+        , [ talo,talon,taloa,talona,taloon,talojen,taloja,taloina,taloissa,taloihin ] ] }
+
+finnishVerbs = Experiment
+  { title = "Finnish verbs"
+  , lexicon = "lib/src/finnish/DictFin.gf"
+  , category = "VK"
+  , nforms        = 12
+  , morphology = "lib/src/finnish/ParadigmsFin.gf"
+  , smartparadigm = "mkV"
+  , setup = \(panna:panen:panee:panevat:pankaa:pannaan:panin:pani:panisi:pannut:pantu:pannee:_) ->
+      map (map esc )
+        [ [ panna ]
+        , [ panna, pani ]
+        , [ panna, panen, pani ]
+        , [ panna, panen, panee, panevat, pankaa, pannaan, panin, pani, panisi, pannut, pantu, pannee ] ]}
+  
+-- ~~~ French ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+frenchVerbs :: Experiment
+frenchVerbs = Experiment
+  { title         = "French verbs"
+  , lexicon       = "lib/src/french/DictFre.gf"
+  , category      = "V"
+  , nforms        = 51
+  , morphology    = "lib/src/french/ParadigmsFre.gf"
+  , smartparadigm = "mkV"
+  , setup         = \vForms ->
+      let jeter = vForms!!0
+          jette = vForms!!3
+          jettera = vForms!!33
+          tenir = vForms!!0
+          tiens = vForms!!1
+          tient = vForms!!3
+          tenons = vForms!!4
+          tenez = vForms!!5
+          tiennent = vForms!!6
+          tienne = vForms!!7
+          tenions = vForms!!10
+          tiensI = vForms!!43
+          tint = vForms!!27
+          tiendra = vForms!!33
+          tenu = vForms!!46
+      in map (map esc )
+        [ [ jeter ]
+        , [ jeter, jette, jettera ]
+        , [ tenir, tiens, tenons, tiennent, tint, tiendra, tenu ]
+        , [ tenir,tiens,tient,tenons,tenez,tiennent,tienne,tenions,tiensI,tint,tiendra,tenu ]
+        ] }
+  -- ~~~ Swedish ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+swedishNouns, swedishVerbs :: Experiment
+swedishNouns = Experiment
+  { title = "Swedish nouns"
+  , lexicon = "lib/src/swedish/DictSwe.gf"
+  , category = "N"
+  , nforms        = 8
+  , morphology = "lib/src/swedish/ParadigmsSwe.gf"
+  , smartparadigm = "mkN"
+  , setup = \(gurka:gurkas:gurkan:gurkans:gurkor:gurkors:gurkorna:gurkornas:_) ->
+      [ [ esc gurka ]
+      , [ esc gurka, esc gurkor ]
+      , [ esc gurka, esc gurkan , esc gurkor, esc gurkorna ] ] }
+
+swedishVerbs = Experiment
+  { title = "Swedish verbs"
+  , lexicon = "lib/src/swedish/DictSwe.gf"
+  , category = "V"
+  , nforms        = 17
+  , morphology = "lib/src/swedish/ParadigmsSwe.gf"
+  , smartparadigm = "mkV"
+  , setup = \(gaar:gaas:gick:gicks:gaa_imp:gaa_inf:gaas_inf:gaatt:gaatts:gaangen:gaangens:gaanget:gaangets:gaangna_undef:gaangnas_undef:gaangna_def:gaangnas_def:_) ->
+      [ [ esc gaa_inf ]
+      , [ esc gaa_inf, esc gick, esc gaatt ]
+      , [ esc gaa_inf, esc gaar, esc gaa_imp, esc gick, esc gaatt, esc gaangen ] ] }