no weights testcase

nod-ai · Oct 1, 2024 · 0fcf2b1 · 0fcf2b1
1 parent 4190e75
commit 0fcf2b1
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/alt_e2eshark/onnx_tests/operators/tfidf_vectorizer.py b/alt_e2eshark/onnx_tests/operators/tfidf_vectorizer.py
@@ -37,11 +37,21 @@
         "min_gram_length": 1,
         "max_gram_length": 2,
         "max_skip_count": 5,
-        "ngram_counts":[0,4],
+        "ngram_counts":[0,1],
         "pool_int64s": [2, 3, 5, 4, 5, 6, 7, 8, 6, 7],
         "ngram_indexes": [0, 2, 4, 5, 6, 7, 8, 9],
         "weights": [0.0, 0.15, 0.1, 1.0, 1.0, 7.0, 2.0, 0.0],
     },
+    "unigram_bigram_no_weights": {
+        "n": 4,
+        "min_gram_length": 1,
+        "max_gram_length": 2,
+        "max_skip_count": 2,
+        "ngram_counts":[0,2],
+        "pool_int64s": [2, 3, 5, 4, 5, 6, 7, 8, 6, 7],
+        "ngram_indexes": [0, 2, 4, 5, 6, 7, 8, 9],
+        "weights": None,
+    },
 }
 
 def create_vectorizer_node(