Merge pull request #336 from vespa-engine/tgm/ranking-framework

Experimental ranking framework
vespa-engine · May 30, 2022 · e2fe0a2 · e2fe0a2
2 parents 99459cb + ae4a9e6
commit e2fe0a2
Show file tree

Hide file tree

Showing 8 changed files with 1,242 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -32,8 +32,8 @@ def get_target_version():
         "tenacity",
     ],
     extras_require={
-        "ml": ["transformers", "torch"],
-        "full": ["transformers", "torch", "onnxruntime"],
+        "ml": ["transformers", "torch", "tensorflow", "tensorflow_ranking", "keras_tuner"],
+        "full": ["onnxruntime", "transformers", "torch", "tensorflow", "tensorflow_ranking", "keras_tuner"],
     },
     python_requires=">=3.6",
     zip_safe=False,

diff --git a/vespa/experimental/__init__.py b/vespa/experimental/__init__.py