fixes word_to_word event file creation for line event structure (#246)

quantling · Nov 28, 2023 · be6c481 · be6c481
1 parent 891197c
commit be6c481
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/pyndl/preprocess.py b/pyndl/preprocess.py
@@ -322,8 +322,12 @@ def gen_occurrences(words):
                 occurrences.append(("_".join(cues), word))
             return occurrences
         elif event_structure == 'line':
-            # (cues, outcomes) with empty outcomes
-            return [('_'.join(words), ''), ]
+            #{"trigrams_to_word", "word_to_word", "bigrams_to_word"}
+            if cue_structure in ("trigrams_to_word", "bigrams_to_word"):
+                # (cues, outcomes) with empty outcomes
+                return [('_'.join(words), ''), ]
+            else:
+                return [('_'.join(words), '_'.join(words)), ]
         else:
             raise ValueError('gen_occurrences should be one of {"consecutive_words", "word_to_word", "line"}')
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pyndl"
-version = "1.1.2"
+version = "1.2.0"
 description = "Naive discriminative learning implements learning and classification models based on the Rescorla-Wagner equations."
 
 license = "MIT"