CSR for variable length data in embedding training (#384)

dmlc · Nov 2, 2018 · 205acce · 205acce
1 parent 8b7d53e
commit 205acce
Show file tree

Hide file tree

Showing 21 changed files with 2,168 additions and 2,112 deletions.
diff --git a/conftest.py b/conftest.py
@@ -34,6 +34,7 @@
 import pytest
 
 
+# * Random seed setup
 def pytest_configure():
     """Pytest configuration hook to help reproduce test segfaults
 
@@ -180,3 +181,9 @@ def test_not_ok_with_random_data():
         logging.info(seed_message)
 
     np.random.set_state(post_test_state)
+
+
+# * Shared test fixtures
+@pytest.fixture(params=[True, False])
+def hybridize(request):
+    return request.param
diff --git a/docs/api/model.train.rst b/docs/api/model.train.rst
@@ -26,7 +26,7 @@ Word Embeddings
     :nosignatures:
 
     EmbeddingModel
-    SimpleEmbeddingModel
+    CSREmbeddingModel
     FasttextEmbeddingModel