diff --git a/_modules/besskge/embedding.html b/_modules/besskge/embedding.html
index e7e8b24..be6f019 100644
--- a/_modules/besskge/embedding.html
+++ b/_modules/besskge/embedding.html
@@ -118,6 +118,22 @@ <h1>Source code for besskge.embedding</h1><div class="highlight"><pre>
     <span class="p">)</span></div>
 
 
+<div class="viewcode-block" id="init_uniform_rotation"><a class="viewcode-back" href="../../generated/besskge.embedding.init_uniform_rotation.html#besskge.embedding.init_uniform_rotation">[docs]</a><span class="k">def</span> <span class="nf">init_uniform_rotation</span><span class="p">(</span><span class="n">embedding_table</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">:</span>
+<span class="w">    </span><span class="sa">r</span><span class="sd">&quot;&quot;&quot;</span>
+<span class="sd">    Initialize tensor with each entry being a uniformly distributed</span>
+<span class="sd">    phase between 0 and :math:`2 \pi`.</span>
+<span class="sd">    To be used for initialization of relation embedding tables</span>
+<span class="sd">    in RotatE scoring function.</span>
+
+<span class="sd">    :param embedding_table:</span>
+<span class="sd">        Tensor of embedding parameters to initialize.</span>
+
+<span class="sd">    :return:</span>
+<span class="sd">        Initialized tensor.</span>
+<span class="sd">    &quot;&quot;&quot;</span>
+    <span class="k">return</span> <span class="n">torch</span><span class="o">.</span><span class="n">rand_like</span><span class="p">(</span><span class="n">embedding_table</span><span class="p">)</span> <span class="o">*</span> <span class="mi">2</span> <span class="o">*</span> <span class="n">np</span><span class="o">.</span><span class="n">pi</span></div>
+
+
 <div class="viewcode-block" id="init_KGE_uniform"><a class="viewcode-back" href="../../generated/besskge.embedding.init_KGE_uniform.html#besskge.embedding.init_KGE_uniform">[docs]</a><span class="k">def</span> <span class="nf">init_KGE_uniform</span><span class="p">(</span>
     <span class="n">embedding_table</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">,</span> <span class="n">b</span><span class="p">:</span> <span class="nb">float</span> <span class="o">=</span> <span class="mf">1.0</span><span class="p">,</span> <span class="n">divide_by_embedding_size</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">True</span>
 <span class="p">)</span> <span class="o">-&gt;</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">:</span>
diff --git a/_modules/besskge/scoring.html b/_modules/besskge/scoring.html
index 9da5b9d..2ad3abc 100644
--- a/_modules/besskge/scoring.html
+++ b/_modules/besskge/scoring.html
@@ -89,6 +89,7 @@ <h1>Source code for besskge.scoring</h1><div class="highlight"><pre>
     <span class="n">init_KGE_normal</span><span class="p">,</span>
     <span class="n">init_KGE_uniform</span><span class="p">,</span>
     <span class="n">init_uniform_norm</span><span class="p">,</span>
+    <span class="n">init_uniform_rotation</span><span class="p">,</span>
     <span class="n">init_xavier_norm</span><span class="p">,</span>
     <span class="n">initialize_entity_embedding</span><span class="p">,</span>
     <span class="n">initialize_relation_embedding</span><span class="p">,</span>
@@ -443,7 +444,7 @@ <h1>Source code for besskge.scoring</h1><div class="highlight"><pre>
             <span class="n">init_KGE_uniform</span>
         <span class="p">],</span>
         <span class="n">relation_initializer</span><span class="p">:</span> <span class="n">Union</span><span class="p">[</span><span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">,</span> <span class="n">List</span><span class="p">[</span><span class="n">Callable</span><span class="p">[</span><span class="o">...</span><span class="p">,</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">]]]</span> <span class="o">=</span> <span class="p">[</span>
-            <span class="n">init_KGE_uniform</span>
+            <span class="n">init_uniform_rotation</span>
         <span class="p">],</span>
         <span class="n">inverse_relations</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span>
     <span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
diff --git a/_sources/generated/besskge.embedding.init_uniform_rotation.rst.txt b/_sources/generated/besskge.embedding.init_uniform_rotation.rst.txt
new file mode 100644
index 0000000..4fee4ca
--- /dev/null
+++ b/_sources/generated/besskge.embedding.init_uniform_rotation.rst.txt
@@ -0,0 +1,6 @@
+besskge.embedding.init\_uniform\_rotation
+=========================================
+
+.. currentmodule:: besskge.embedding
+
+.. autofunction:: init_uniform_rotation
\ No newline at end of file
diff --git a/_sources/generated/besskge.embedding.rst.txt b/_sources/generated/besskge.embedding.rst.txt
index 901b5be..82b8427 100644
--- a/_sources/generated/besskge.embedding.rst.txt
+++ b/_sources/generated/besskge.embedding.rst.txt
@@ -21,6 +21,7 @@ besskge.embedding
       init_KGE_normal
       init_KGE_uniform
       init_uniform_norm
+      init_uniform_rotation
       init_xavier_norm
       initialize_entity_embedding
       initialize_relation_embedding
diff --git a/generated/besskge.dataset.KGDataset.html b/generated/besskge.dataset.KGDataset.html
index 2c05df2..0fcaad9 100644
--- a/generated/besskge.dataset.KGDataset.html
+++ b/generated/besskge.dataset.KGDataset.html
@@ -223,12 +223,12 @@ <h1>besskge.dataset.KGDataset<a class="headerlink" href="#besskge-dataset-kgdata
 <dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
 <dd class="field-odd"><ul class="simple">
-<li><p><strong>df</strong> (<a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Union" title="(in Python v3.12)"><code class="xref py py-data docutils literal notranslate"><span class="pre">Union</span></code></a>[<a class="reference external" href="http://pandas.pydata.org/pandas-docs/dev/reference/api/pandas.DataFrame.html#pandas.DataFrame" title="(in pandas v2.3.0.dev0+54.g04b45b10b1)"><code class="xref py py-class docutils literal notranslate"><span class="pre">DataFrame</span></code></a>, <a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Dict" title="(in Python v3.12)"><code class="xref py py-class docutils literal notranslate"><span class="pre">Dict</span></code></a>[<a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.12)"><code class="xref py py-class docutils literal notranslate"><span class="pre">str</span></code></a>, <a class="reference external" href="http://pandas.pydata.org/pandas-docs/dev/reference/api/pandas.DataFrame.html#pandas.DataFrame" title="(in pandas v2.3.0.dev0+54.g04b45b10b1)"><code class="xref py py-class docutils literal notranslate"><span class="pre">DataFrame</span></code></a>]]) – Pandas DataFrame of all triples in the knowledge graph dataset,
+<li><p><strong>df</strong> (<a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Union" title="(in Python v3.12)"><code class="xref py py-data docutils literal notranslate"><span class="pre">Union</span></code></a>[<a class="reference external" href="http://pandas.pydata.org/pandas-docs/dev/reference/api/pandas.DataFrame.html#pandas.DataFrame" title="(in pandas v3.0.0.dev0+163.g7281475054)"><code class="xref py py-class docutils literal notranslate"><span class="pre">DataFrame</span></code></a>, <a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Dict" title="(in Python v3.12)"><code class="xref py py-class docutils literal notranslate"><span class="pre">Dict</span></code></a>[<a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.12)"><code class="xref py py-class docutils literal notranslate"><span class="pre">str</span></code></a>, <a class="reference external" href="http://pandas.pydata.org/pandas-docs/dev/reference/api/pandas.DataFrame.html#pandas.DataFrame" title="(in pandas v3.0.0.dev0+163.g7281475054)"><code class="xref py py-class docutils literal notranslate"><span class="pre">DataFrame</span></code></a>]]) – Pandas DataFrame of all triples in the knowledge graph dataset,
 or dictionary of DataFrames of triples for each part of the dataset split</p></li>
 <li><p><strong>head_column</strong> (<a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Union" title="(in Python v3.12)"><code class="xref py py-data docutils literal notranslate"><span class="pre">Union</span></code></a>[<a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.12)"><code class="xref py py-class docutils literal notranslate"><span class="pre">int</span></code></a>, <a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.12)"><code class="xref py py-class docutils literal notranslate"><span class="pre">str</span></code></a>]) – Name of the DataFrame column storing head entities</p></li>
 <li><p><strong>relation_column</strong> (<a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Union" title="(in Python v3.12)"><code class="xref py py-data docutils literal notranslate"><span class="pre">Union</span></code></a>[<a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.12)"><code class="xref py py-class docutils literal notranslate"><span class="pre">int</span></code></a>, <a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.12)"><code class="xref py py-class docutils literal notranslate"><span class="pre">str</span></code></a>]) – Name of the DataFrame column storing relations</p></li>
 <li><p><strong>tail_column</strong> (<a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Union" title="(in Python v3.12)"><code class="xref py py-data docutils literal notranslate"><span class="pre">Union</span></code></a>[<a class="reference external" href="https://docs.python.org/3/library/functions.html#int" title="(in Python v3.12)"><code class="xref py py-class docutils literal notranslate"><span class="pre">int</span></code></a>, <a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.12)"><code class="xref py py-class docutils literal notranslate"><span class="pre">str</span></code></a>]) – Name of the DataFrame column storing tail entities</p></li>
-<li><p><strong>entity_types</strong> (<a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Union" title="(in Python v3.12)"><code class="xref py py-data docutils literal notranslate"><span class="pre">Union</span></code></a>[<a class="reference external" href="http://pandas.pydata.org/pandas-docs/dev/reference/api/pandas.Series.html#pandas.Series" title="(in pandas v2.3.0.dev0+54.g04b45b10b1)"><code class="xref py py-class docutils literal notranslate"><span class="pre">Series</span></code></a>, <a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Dict" title="(in Python v3.12)"><code class="xref py py-class docutils literal notranslate"><span class="pre">Dict</span></code></a>[<a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.12)"><code class="xref py py-class docutils literal notranslate"><span class="pre">str</span></code></a>, <a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.12)"><code class="xref py py-class docutils literal notranslate"><span class="pre">str</span></code></a>], <a class="reference external" href="https://docs.python.org/3/library/constants.html#None" title="(in Python v3.12)"><code class="xref py py-obj docutils literal notranslate"><span class="pre">None</span></code></a>]) – If entities have types, dictionary or pandas Series of mappings
+<li><p><strong>entity_types</strong> (<a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Union" title="(in Python v3.12)"><code class="xref py py-data docutils literal notranslate"><span class="pre">Union</span></code></a>[<a class="reference external" href="http://pandas.pydata.org/pandas-docs/dev/reference/api/pandas.Series.html#pandas.Series" title="(in pandas v3.0.0.dev0+163.g7281475054)"><code class="xref py py-class docutils literal notranslate"><span class="pre">Series</span></code></a>, <a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Dict" title="(in Python v3.12)"><code class="xref py py-class docutils literal notranslate"><span class="pre">Dict</span></code></a>[<a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.12)"><code class="xref py py-class docutils literal notranslate"><span class="pre">str</span></code></a>, <a class="reference external" href="https://docs.python.org/3/library/stdtypes.html#str" title="(in Python v3.12)"><code class="xref py py-class docutils literal notranslate"><span class="pre">str</span></code></a>], <a class="reference external" href="https://docs.python.org/3/library/constants.html#None" title="(in Python v3.12)"><code class="xref py py-obj docutils literal notranslate"><span class="pre">None</span></code></a>]) – If entities have types, dictionary or pandas Series of mappings
 entity label -&gt; entity type (as strings).</p></li>
 <li><p><strong>split</strong> (<a class="reference external" href="https://docs.python.org/3/library/typing.html#typing.Tuple" title="(in Python v3.12)"><code class="xref py py-data docutils literal notranslate"><span class="pre">Tuple</span></code></a>[<a class="reference external" href="https://docs.python.org/3/library/functions.html#float" title="(in Python v3.12)"><code class="xref py py-class docutils literal notranslate"><span class="pre">float</span></code></a>, <a class="reference external" href="https://docs.python.org/3/library/functions.html#float" title="(in Python v3.12)"><code class="xref py py-class docutils literal notranslate"><span class="pre">float</span></code></a>, <a class="reference external" href="https://docs.python.org/3/library/functions.html#float" title="(in Python v3.12)"><code class="xref py py-class docutils literal notranslate"><span class="pre">float</span></code></a>]) – Tuple to set the train/validation/test split.
 Only used if no pre-defined dataset split is specified,
diff --git a/generated/besskge.embedding.html b/generated/besskge.embedding.html
index 5a9e0b8..62bd4dc 100644
--- a/generated/besskge.embedding.html
+++ b/generated/besskge.embedding.html
@@ -15,6 +15,7 @@
         <script data-url_root="../" id="documentation_options" src="../_static/documentation_options.js"></script>
         <script src="../_static/doctools.js"></script>
         <script src="../_static/sphinx_highlight.js"></script>
+        <script async="async" src="https://cdn.jsdelivr.net/npm/mathjax@3/es5/tex-mml-chtml.js"></script>
     <script src="../_static/js/theme.js"></script>
     <link rel="index" title="Index" href="../genindex.html" />
     <link rel="search" title="Search" href="../search.html" />
@@ -59,6 +60,7 @@
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_KGE_normal.html">besskge.embedding.init_KGE_normal</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_KGE_uniform.html">besskge.embedding.init_KGE_uniform</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_uniform_norm.html">besskge.embedding.init_uniform_norm</a></li>
+<li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_uniform_rotation.html">besskge.embedding.init_uniform_rotation</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_xavier_norm.html">besskge.embedding.init_xavier_norm</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.initialize_entity_embedding.html">besskge.embedding.initialize_entity_embedding</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.initialize_relation_embedding.html">besskge.embedding.initialize_relation_embedding</a></li>
@@ -112,16 +114,19 @@
 <tr class="row-odd"><td><p><a class="reference internal" href="besskge.embedding.init_uniform_norm.html#besskge.embedding.init_uniform_norm" title="besskge.embedding.init_uniform_norm"><code class="xref py py-obj docutils literal notranslate"><span class="pre">init_uniform_norm</span></code></a>(embedding_table)</p></td>
 <td><p>Initialize embeddings according to uniform distribution and normalize so that each row has norm 1.</p></td>
 </tr>
-<tr class="row-even"><td><p><a class="reference internal" href="besskge.embedding.init_xavier_norm.html#besskge.embedding.init_xavier_norm" title="besskge.embedding.init_xavier_norm"><code class="xref py py-obj docutils literal notranslate"><span class="pre">init_xavier_norm</span></code></a>(embedding_table[, gain])</p></td>
+<tr class="row-even"><td><p><a class="reference internal" href="besskge.embedding.init_uniform_rotation.html#besskge.embedding.init_uniform_rotation" title="besskge.embedding.init_uniform_rotation"><code class="xref py py-obj docutils literal notranslate"><span class="pre">init_uniform_rotation</span></code></a>(embedding_table)</p></td>
+<td><p>Initialize tensor with each entry being a uniformly distributed phase between 0 and <span class="math notranslate nohighlight">\(2 \pi\)</span>.</p></td>
+</tr>
+<tr class="row-odd"><td><p><a class="reference internal" href="besskge.embedding.init_xavier_norm.html#besskge.embedding.init_xavier_norm" title="besskge.embedding.init_xavier_norm"><code class="xref py py-obj docutils literal notranslate"><span class="pre">init_xavier_norm</span></code></a>(embedding_table[, gain])</p></td>
 <td><p>Initialize embeddings according to Xavier normal scheme, with <cite>fan_in = 0</cite>, <cite>fan_out=row_size</cite>.</p></td>
 </tr>
-<tr class="row-odd"><td><p><a class="reference internal" href="besskge.embedding.initialize_entity_embedding.html#besskge.embedding.initialize_entity_embedding" title="besskge.embedding.initialize_entity_embedding"><code class="xref py py-obj docutils literal notranslate"><span class="pre">initialize_entity_embedding</span></code></a>(sharding, ...[, ...])</p></td>
+<tr class="row-even"><td><p><a class="reference internal" href="besskge.embedding.initialize_entity_embedding.html#besskge.embedding.initialize_entity_embedding" title="besskge.embedding.initialize_entity_embedding"><code class="xref py py-obj docutils literal notranslate"><span class="pre">initialize_entity_embedding</span></code></a>(sharding, ...[, ...])</p></td>
 <td><p>Initialize entity embedding table.</p></td>
 </tr>
-<tr class="row-even"><td><p><a class="reference internal" href="besskge.embedding.initialize_relation_embedding.html#besskge.embedding.initialize_relation_embedding" title="besskge.embedding.initialize_relation_embedding"><code class="xref py py-obj docutils literal notranslate"><span class="pre">initialize_relation_embedding</span></code></a>(...[, row_size])</p></td>
+<tr class="row-odd"><td><p><a class="reference internal" href="besskge.embedding.initialize_relation_embedding.html#besskge.embedding.initialize_relation_embedding" title="besskge.embedding.initialize_relation_embedding"><code class="xref py py-obj docutils literal notranslate"><span class="pre">initialize_relation_embedding</span></code></a>(...[, row_size])</p></td>
 <td><p>Initialize relation embedding table.</p></td>
 </tr>
-<tr class="row-odd"><td><p><a class="reference internal" href="besskge.embedding.refactor_embedding_sharding.html#besskge.embedding.refactor_embedding_sharding" title="besskge.embedding.refactor_embedding_sharding"><code class="xref py py-obj docutils literal notranslate"><span class="pre">refactor_embedding_sharding</span></code></a>(...)</p></td>
+<tr class="row-even"><td><p><a class="reference internal" href="besskge.embedding.refactor_embedding_sharding.html#besskge.embedding.refactor_embedding_sharding" title="besskge.embedding.refactor_embedding_sharding"><code class="xref py py-obj docutils literal notranslate"><span class="pre">refactor_embedding_sharding</span></code></a>(...)</p></td>
 <td><p>Refactor sharded entity embedding table to pass from one entity sharding to a different one.</p></td>
 </tr>
 </tbody>
diff --git a/generated/besskge.embedding.init_KGE_normal.html b/generated/besskge.embedding.init_KGE_normal.html
index 0035a49..55640b3 100644
--- a/generated/besskge.embedding.init_KGE_normal.html
+++ b/generated/besskge.embedding.init_KGE_normal.html
@@ -62,6 +62,7 @@
 </li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_KGE_uniform.html">besskge.embedding.init_KGE_uniform</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_uniform_norm.html">besskge.embedding.init_uniform_norm</a></li>
+<li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_uniform_rotation.html">besskge.embedding.init_uniform_rotation</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_xavier_norm.html">besskge.embedding.init_xavier_norm</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.initialize_entity_embedding.html">besskge.embedding.initialize_entity_embedding</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.initialize_relation_embedding.html">besskge.embedding.initialize_relation_embedding</a></li>
diff --git a/generated/besskge.embedding.init_KGE_uniform.html b/generated/besskge.embedding.init_KGE_uniform.html
index 4d0f1e7..ab2cb98 100644
--- a/generated/besskge.embedding.init_KGE_uniform.html
+++ b/generated/besskge.embedding.init_KGE_uniform.html
@@ -62,6 +62,7 @@
 </ul>
 </li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_uniform_norm.html">besskge.embedding.init_uniform_norm</a></li>
+<li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_uniform_rotation.html">besskge.embedding.init_uniform_rotation</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_xavier_norm.html">besskge.embedding.init_xavier_norm</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.initialize_entity_embedding.html">besskge.embedding.initialize_entity_embedding</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.initialize_relation_embedding.html">besskge.embedding.initialize_relation_embedding</a></li>
diff --git a/generated/besskge.embedding.init_uniform_norm.html b/generated/besskge.embedding.init_uniform_norm.html
index 036d1b5..be1bd18 100644
--- a/generated/besskge.embedding.init_uniform_norm.html
+++ b/generated/besskge.embedding.init_uniform_norm.html
@@ -18,7 +18,7 @@
     <script src="../_static/js/theme.js"></script>
     <link rel="index" title="Index" href="../genindex.html" />
     <link rel="search" title="Search" href="../search.html" />
-    <link rel="next" title="besskge.embedding.init_xavier_norm" href="besskge.embedding.init_xavier_norm.html" />
+    <link rel="next" title="besskge.embedding.init_uniform_rotation" href="besskge.embedding.init_uniform_rotation.html" />
     <link rel="prev" title="besskge.embedding.init_KGE_uniform" href="besskge.embedding.init_KGE_uniform.html" /> 
 </head>
 
@@ -62,6 +62,7 @@
 <li class="toctree-l4"><a class="reference internal" href="#besskge.embedding.init_uniform_norm"><code class="docutils literal notranslate"><span class="pre">init_uniform_norm()</span></code></a></li>
 </ul>
 </li>
+<li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_uniform_rotation.html">besskge.embedding.init_uniform_rotation</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_xavier_norm.html">besskge.embedding.init_xavier_norm</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.initialize_entity_embedding.html">besskge.embedding.initialize_entity_embedding</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.initialize_relation_embedding.html">besskge.embedding.initialize_relation_embedding</a></li>
@@ -128,7 +129,7 @@ <h1>besskge.embedding.init_uniform_norm<a class="headerlink" href="#besskge-embe
           </div>
           <footer><div class="rst-footer-buttons" role="navigation" aria-label="Footer">
         <a href="besskge.embedding.init_KGE_uniform.html" class="btn btn-neutral float-left" title="besskge.embedding.init_KGE_uniform" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
-        <a href="besskge.embedding.init_xavier_norm.html" class="btn btn-neutral float-right" title="besskge.embedding.init_xavier_norm" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
+        <a href="besskge.embedding.init_uniform_rotation.html" class="btn btn-neutral float-right" title="besskge.embedding.init_uniform_rotation" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
     </div>
 
   <hr/>
diff --git a/generated/besskge.embedding.init_uniform_rotation.html b/generated/besskge.embedding.init_uniform_rotation.html
new file mode 100644
index 0000000..5c8b071
--- /dev/null
+++ b/generated/besskge.embedding.init_uniform_rotation.html
@@ -0,0 +1,161 @@
+<!DOCTYPE html>
+<html class="writer-html5" lang="en" >
+<head>
+  <meta charset="utf-8" /><meta name="generator" content="Docutils 0.18.1: http://docutils.sourceforge.net/" />
+
+  <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+  <title>besskge.embedding.init_uniform_rotation &mdash; BESS-KGE  documentation</title>
+      <link rel="stylesheet" href="../_static/pygments.css" type="text/css" />
+      <link rel="stylesheet" href="../_static/css/theme.css" type="text/css" />
+      <link rel="stylesheet" href="../_static/graphviz.css" type="text/css" />
+  <!--[if lt IE 9]>
+    <script src="../_static/js/html5shiv.min.js"></script>
+  <![endif]-->
+  
+        <script data-url_root="../" id="documentation_options" src="../_static/documentation_options.js"></script>
+        <script src="../_static/doctools.js"></script>
+        <script src="../_static/sphinx_highlight.js"></script>
+        <script async="async" src="https://cdn.jsdelivr.net/npm/mathjax@3/es5/tex-mml-chtml.js"></script>
+    <script src="../_static/js/theme.js"></script>
+    <link rel="index" title="Index" href="../genindex.html" />
+    <link rel="search" title="Search" href="../search.html" />
+    <link rel="next" title="besskge.embedding.init_xavier_norm" href="besskge.embedding.init_xavier_norm.html" />
+    <link rel="prev" title="besskge.embedding.init_uniform_norm" href="besskge.embedding.init_uniform_norm.html" /> 
+</head>
+
+<body class="wy-body-for-nav"> 
+  <div class="wy-grid-for-nav">
+    <nav data-toggle="wy-nav-shift" class="wy-nav-side">
+      <div class="wy-side-scroll">
+        <div class="wy-side-nav-search" >
+
+          
+          
+          <a href="../index.html" class="icon icon-home">
+            BESS-KGE
+          </a>
+<div role="search">
+  <form id="rtd-search-form" class="wy-form" action="../search.html" method="get">
+    <input type="text" name="q" placeholder="Search docs" aria-label="Search docs" />
+    <input type="hidden" name="check_keywords" value="yes" />
+    <input type="hidden" name="area" value="default" />
+  </form>
+</div>
+        </div><div class="wy-menu wy-menu-vertical" data-spy="affix" role="navigation" aria-label="Navigation menu">
+              <p class="caption" role="heading"><span class="caption-text">Contents</span></p>
+<ul class="current">
+<li class="toctree-l1"><a class="reference internal" href="../user_guide.html">User guide</a></li>
+<li class="toctree-l1"><a class="reference internal" href="../bess.html">BESS overview</a></li>
+<li class="toctree-l1 current"><a class="reference internal" href="../API_reference.html">API reference</a><ul class="current">
+<li class="toctree-l2"><a class="reference internal" href="besskge.pipeline.html">besskge.pipeline</a></li>
+<li class="toctree-l2"><a class="reference internal" href="besskge.dataset.html">besskge.dataset</a></li>
+<li class="toctree-l2"><a class="reference internal" href="besskge.sharding.html">besskge.sharding</a></li>
+<li class="toctree-l2"><a class="reference internal" href="besskge.batch_sampler.html">besskge.batch_sampler</a></li>
+<li class="toctree-l2"><a class="reference internal" href="besskge.negative_sampler.html">besskge.negative_sampler</a></li>
+<li class="toctree-l2"><a class="reference internal" href="besskge.bess.html">besskge.bess</a></li>
+<li class="toctree-l2"><a class="reference internal" href="besskge.scoring.html">besskge.scoring</a></li>
+<li class="toctree-l2"><a class="reference internal" href="besskge.loss.html">besskge.loss</a></li>
+<li class="toctree-l2"><a class="reference internal" href="besskge.metric.html">besskge.metric</a></li>
+<li class="toctree-l2 current"><a class="reference internal" href="besskge.embedding.html">besskge.embedding</a><ul class="current">
+<li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_KGE_normal.html">besskge.embedding.init_KGE_normal</a></li>
+<li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_KGE_uniform.html">besskge.embedding.init_KGE_uniform</a></li>
+<li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_uniform_norm.html">besskge.embedding.init_uniform_norm</a></li>
+<li class="toctree-l3 current"><a class="current reference internal" href="#">besskge.embedding.init_uniform_rotation</a><ul>
+<li class="toctree-l4"><a class="reference internal" href="#besskge.embedding.init_uniform_rotation"><code class="docutils literal notranslate"><span class="pre">init_uniform_rotation()</span></code></a></li>
+</ul>
+</li>
+<li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_xavier_norm.html">besskge.embedding.init_xavier_norm</a></li>
+<li class="toctree-l3"><a class="reference internal" href="besskge.embedding.initialize_entity_embedding.html">besskge.embedding.initialize_entity_embedding</a></li>
+<li class="toctree-l3"><a class="reference internal" href="besskge.embedding.initialize_relation_embedding.html">besskge.embedding.initialize_relation_embedding</a></li>
+<li class="toctree-l3"><a class="reference internal" href="besskge.embedding.refactor_embedding_sharding.html">besskge.embedding.refactor_embedding_sharding</a></li>
+</ul>
+</li>
+<li class="toctree-l2"><a class="reference internal" href="besskge.utils.html">besskge.utils</a></li>
+</ul>
+</li>
+<li class="toctree-l1"><a class="reference internal" href="../dev_guide.html">Developers guide</a></li>
+<li class="toctree-l1"><a class="reference internal" href="../bibliography.html">Bibliography</a></li>
+</ul>
+
+        </div>
+      </div>
+    </nav>
+
+    <section data-toggle="wy-nav-shift" class="wy-nav-content-wrap"><nav class="wy-nav-top" aria-label="Mobile navigation menu" >
+          <i data-toggle="wy-nav-top" class="fa fa-bars"></i>
+          <a href="../index.html">BESS-KGE</a>
+      </nav>
+
+      <div class="wy-nav-content">
+        <div class="rst-content">
+          <div role="navigation" aria-label="Page navigation">
+  <ul class="wy-breadcrumbs">
+      <li><a href="../index.html" class="icon icon-home" aria-label="Home"></a></li>
+          <li class="breadcrumb-item"><a href="../API_reference.html">BESS-KGE API Reference</a></li>
+          <li class="breadcrumb-item"><a href="besskge.embedding.html">besskge.embedding</a></li>
+      <li class="breadcrumb-item active">besskge.embedding.init_uniform_rotation</li>
+      <li class="wy-breadcrumbs-aside">
+            <a href="../_sources/generated/besskge.embedding.init_uniform_rotation.rst.txt" rel="nofollow"> View page source</a>
+      </li>
+  </ul>
+  <hr/>
+</div>
+          <div role="main" class="document" itemscope="itemscope" itemtype="http://schema.org/Article">
+           <div itemprop="articleBody">
+             
+  <section id="besskge-embedding-init-uniform-rotation">
+<h1>besskge.embedding.init_uniform_rotation<a class="headerlink" href="#besskge-embedding-init-uniform-rotation" title="Permalink to this heading"></a></h1>
+<dl class="py function">
+<dt class="sig sig-object py" id="besskge.embedding.init_uniform_rotation">
+<span class="sig-prename descclassname"><span class="pre">besskge.embedding.</span></span><span class="sig-name descname"><span class="pre">init_uniform_rotation</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">embedding_table</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/besskge/embedding.html#init_uniform_rotation"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#besskge.embedding.init_uniform_rotation" title="Permalink to this definition"></a></dt>
+<dd><p>Initialize tensor with each entry being a uniformly distributed
+phase between 0 and <span class="math notranslate nohighlight">\(2 \pi\)</span>.
+To be used for initialization of relation embedding tables
+in RotatE scoring function.</p>
+<dl class="field-list simple">
+<dt class="field-odd">Parameters<span class="colon">:</span></dt>
+<dd class="field-odd"><p><strong>embedding_table</strong> (<a class="reference external" href="https://pytorch.org/docs/stable/tensors.html#torch.Tensor" title="(in PyTorch v2.1)"><code class="xref py py-class docutils literal notranslate"><span class="pre">Tensor</span></code></a>) – Tensor of embedding parameters to initialize.</p>
+</dd>
+<dt class="field-even">Return type<span class="colon">:</span></dt>
+<dd class="field-even"><p><a class="reference external" href="https://pytorch.org/docs/stable/tensors.html#torch.Tensor" title="(in PyTorch v2.1)"><code class="xref py py-class docutils literal notranslate"><span class="pre">Tensor</span></code></a></p>
+</dd>
+<dt class="field-odd">Returns<span class="colon">:</span></dt>
+<dd class="field-odd"><p>Initialized tensor.</p>
+</dd>
+</dl>
+</dd></dl>
+
+</section>
+
+
+           </div>
+          </div>
+          <footer><div class="rst-footer-buttons" role="navigation" aria-label="Footer">
+        <a href="besskge.embedding.init_uniform_norm.html" class="btn btn-neutral float-left" title="besskge.embedding.init_uniform_norm" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
+        <a href="besskge.embedding.init_xavier_norm.html" class="btn btn-neutral float-right" title="besskge.embedding.init_xavier_norm" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
+    </div>
+
+  <hr/>
+
+  <div role="contentinfo">
+    <p>&#169; Copyright (c) 2023 Graphcore Ltd. All rights reserved.</p>
+  </div>
+
+  Built with <a href="https://www.sphinx-doc.org/">Sphinx</a> using a
+    <a href="https://github.com/readthedocs/sphinx_rtd_theme">theme</a>
+    provided by <a href="https://readthedocs.org">Read the Docs</a>.
+   
+
+</footer>
+        </div>
+      </div>
+    </section>
+  </div>
+  <script>
+      jQuery(function () {
+          SphinxRtdTheme.Navigation.enable(true);
+      });
+  </script> 
+
+</body>
+</html>
\ No newline at end of file
diff --git a/generated/besskge.embedding.init_xavier_norm.html b/generated/besskge.embedding.init_xavier_norm.html
index 4d22e0f..6cd16c1 100644
--- a/generated/besskge.embedding.init_xavier_norm.html
+++ b/generated/besskge.embedding.init_xavier_norm.html
@@ -19,7 +19,7 @@
     <link rel="index" title="Index" href="../genindex.html" />
     <link rel="search" title="Search" href="../search.html" />
     <link rel="next" title="besskge.embedding.initialize_entity_embedding" href="besskge.embedding.initialize_entity_embedding.html" />
-    <link rel="prev" title="besskge.embedding.init_uniform_norm" href="besskge.embedding.init_uniform_norm.html" /> 
+    <link rel="prev" title="besskge.embedding.init_uniform_rotation" href="besskge.embedding.init_uniform_rotation.html" /> 
 </head>
 
 <body class="wy-body-for-nav"> 
@@ -59,6 +59,7 @@
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_KGE_normal.html">besskge.embedding.init_KGE_normal</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_KGE_uniform.html">besskge.embedding.init_KGE_uniform</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_uniform_norm.html">besskge.embedding.init_uniform_norm</a></li>
+<li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_uniform_rotation.html">besskge.embedding.init_uniform_rotation</a></li>
 <li class="toctree-l3 current"><a class="current reference internal" href="#">besskge.embedding.init_xavier_norm</a><ul>
 <li class="toctree-l4"><a class="reference internal" href="#besskge.embedding.init_xavier_norm"><code class="docutils literal notranslate"><span class="pre">init_xavier_norm()</span></code></a></li>
 </ul>
@@ -130,7 +131,7 @@ <h1>besskge.embedding.init_xavier_norm<a class="headerlink" href="#besskge-embed
            </div>
           </div>
           <footer><div class="rst-footer-buttons" role="navigation" aria-label="Footer">
-        <a href="besskge.embedding.init_uniform_norm.html" class="btn btn-neutral float-left" title="besskge.embedding.init_uniform_norm" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
+        <a href="besskge.embedding.init_uniform_rotation.html" class="btn btn-neutral float-left" title="besskge.embedding.init_uniform_rotation" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
         <a href="besskge.embedding.initialize_entity_embedding.html" class="btn btn-neutral float-right" title="besskge.embedding.initialize_entity_embedding" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
     </div>
 
diff --git a/generated/besskge.embedding.initialize_entity_embedding.html b/generated/besskge.embedding.initialize_entity_embedding.html
index a97967b..5ce5aa1 100644
--- a/generated/besskge.embedding.initialize_entity_embedding.html
+++ b/generated/besskge.embedding.initialize_entity_embedding.html
@@ -59,6 +59,7 @@
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_KGE_normal.html">besskge.embedding.init_KGE_normal</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_KGE_uniform.html">besskge.embedding.init_KGE_uniform</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_uniform_norm.html">besskge.embedding.init_uniform_norm</a></li>
+<li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_uniform_rotation.html">besskge.embedding.init_uniform_rotation</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_xavier_norm.html">besskge.embedding.init_xavier_norm</a></li>
 <li class="toctree-l3 current"><a class="current reference internal" href="#">besskge.embedding.initialize_entity_embedding</a><ul>
 <li class="toctree-l4"><a class="reference internal" href="#besskge.embedding.initialize_entity_embedding"><code class="docutils literal notranslate"><span class="pre">initialize_entity_embedding()</span></code></a></li>
diff --git a/generated/besskge.embedding.initialize_relation_embedding.html b/generated/besskge.embedding.initialize_relation_embedding.html
index e137d13..d5a5508 100644
--- a/generated/besskge.embedding.initialize_relation_embedding.html
+++ b/generated/besskge.embedding.initialize_relation_embedding.html
@@ -59,6 +59,7 @@
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_KGE_normal.html">besskge.embedding.init_KGE_normal</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_KGE_uniform.html">besskge.embedding.init_KGE_uniform</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_uniform_norm.html">besskge.embedding.init_uniform_norm</a></li>
+<li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_uniform_rotation.html">besskge.embedding.init_uniform_rotation</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_xavier_norm.html">besskge.embedding.init_xavier_norm</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.initialize_entity_embedding.html">besskge.embedding.initialize_entity_embedding</a></li>
 <li class="toctree-l3 current"><a class="current reference internal" href="#">besskge.embedding.initialize_relation_embedding</a><ul>
diff --git a/generated/besskge.embedding.refactor_embedding_sharding.html b/generated/besskge.embedding.refactor_embedding_sharding.html
index 4479e3d..5fb9ca8 100644
--- a/generated/besskge.embedding.refactor_embedding_sharding.html
+++ b/generated/besskge.embedding.refactor_embedding_sharding.html
@@ -59,6 +59,7 @@
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_KGE_normal.html">besskge.embedding.init_KGE_normal</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_KGE_uniform.html">besskge.embedding.init_KGE_uniform</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_uniform_norm.html">besskge.embedding.init_uniform_norm</a></li>
+<li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_uniform_rotation.html">besskge.embedding.init_uniform_rotation</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.init_xavier_norm.html">besskge.embedding.init_xavier_norm</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.initialize_entity_embedding.html">besskge.embedding.initialize_entity_embedding</a></li>
 <li class="toctree-l3"><a class="reference internal" href="besskge.embedding.initialize_relation_embedding.html">besskge.embedding.initialize_relation_embedding</a></li>
diff --git a/generated/besskge.scoring.RotatE.html b/generated/besskge.scoring.RotatE.html
index 4d010d4..68c666a 100644
--- a/generated/besskge.scoring.RotatE.html
+++ b/generated/besskge.scoring.RotatE.html
@@ -111,7 +111,7 @@
 <h1>besskge.scoring.RotatE<a class="headerlink" href="#besskge-scoring-rotate" title="Permalink to this heading"></a></h1>
 <dl class="py class">
 <dt class="sig sig-object py" id="besskge.scoring.RotatE">
-<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">besskge.scoring.</span></span><span class="sig-name descname"><span class="pre">RotatE</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="pre">negative_sample_sharing,</span> <span class="pre">scoring_norm,</span> <span class="pre">sharding,</span> <span class="pre">n_relation_type,</span> <span class="pre">embedding_size,</span> <span class="pre">entity_initializer=[&lt;function</span> <span class="pre">init_KGE_uniform&gt;],</span> <span class="pre">relation_initializer=[&lt;function</span> <span class="pre">init_KGE_uniform&gt;],</span> <span class="pre">inverse_relations=False</span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/besskge/scoring.html#RotatE"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#besskge.scoring.RotatE" title="Permalink to this definition"></a></dt>
+<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">besskge.scoring.</span></span><span class="sig-name descname"><span class="pre">RotatE</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="pre">negative_sample_sharing,</span> <span class="pre">scoring_norm,</span> <span class="pre">sharding,</span> <span class="pre">n_relation_type,</span> <span class="pre">embedding_size,</span> <span class="pre">entity_initializer=[&lt;function</span> <span class="pre">init_KGE_uniform&gt;],</span> <span class="pre">relation_initializer=[&lt;function</span> <span class="pre">init_uniform_rotation&gt;],</span> <span class="pre">inverse_relations=False</span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/besskge/scoring.html#RotatE"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#besskge.scoring.RotatE" title="Permalink to this definition"></a></dt>
 <dd><p>RotatE scoring function <span id="id1">[<a class="reference internal" href="../bibliography.html#id5" title="Zhiqing Sun, Zhi-Hong Deng, Jian-Yun Nie, and Jian Tang. RotatE: knowledge graph embedding by relational rotation in complex space. In 7th International Conference on Learning Representations, ICLR 2019. 2019.">SDNT19</a>]</span>.</p>
 <p>Initialize RotatE model.</p>
 <dl class="field-list simple">
diff --git a/genindex.html b/genindex.html
index 8e1a1e6..72c2670 100644
--- a/genindex.html
+++ b/genindex.html
@@ -483,10 +483,12 @@ <h2 id="I">I</h2>
 </li>
       <li><a href="generated/besskge.embedding.init_uniform_norm.html#besskge.embedding.init_uniform_norm">init_uniform_norm() (in module besskge.embedding)</a>
 </li>
-      <li><a href="generated/besskge.embedding.init_xavier_norm.html#besskge.embedding.init_xavier_norm">init_xavier_norm() (in module besskge.embedding)</a>
+      <li><a href="generated/besskge.embedding.init_uniform_rotation.html#besskge.embedding.init_uniform_rotation">init_uniform_rotation() (in module besskge.embedding)</a>
 </li>
   </ul></td>
   <td style="width: 33%; vertical-align: top;"><ul>
+      <li><a href="generated/besskge.embedding.init_xavier_norm.html#besskge.embedding.init_xavier_norm">init_xavier_norm() (in module besskge.embedding)</a>
+</li>
       <li><a href="generated/besskge.embedding.initialize_entity_embedding.html#besskge.embedding.initialize_entity_embedding">initialize_entity_embedding() (in module besskge.embedding)</a>
 </li>
       <li><a href="generated/besskge.embedding.initialize_relation_embedding.html#besskge.embedding.initialize_relation_embedding">initialize_relation_embedding() (in module besskge.embedding)</a>
diff --git a/index.html b/index.html
index 3a13deb..fab5bc4 100644
--- a/index.html
+++ b/index.html
@@ -174,6 +174,7 @@
 <li class="toctree-l3"><a class="reference internal" href="generated/besskge.embedding.init_KGE_normal.html">besskge.embedding.init_KGE_normal</a></li>
 <li class="toctree-l3"><a class="reference internal" href="generated/besskge.embedding.init_KGE_uniform.html">besskge.embedding.init_KGE_uniform</a></li>
 <li class="toctree-l3"><a class="reference internal" href="generated/besskge.embedding.init_uniform_norm.html">besskge.embedding.init_uniform_norm</a></li>
+<li class="toctree-l3"><a class="reference internal" href="generated/besskge.embedding.init_uniform_rotation.html">besskge.embedding.init_uniform_rotation</a></li>
 <li class="toctree-l3"><a class="reference internal" href="generated/besskge.embedding.init_xavier_norm.html">besskge.embedding.init_xavier_norm</a></li>
 <li class="toctree-l3"><a class="reference internal" href="generated/besskge.embedding.initialize_entity_embedding.html">besskge.embedding.initialize_entity_embedding</a></li>
 <li class="toctree-l3"><a class="reference internal" href="generated/besskge.embedding.initialize_relation_embedding.html">besskge.embedding.initialize_relation_embedding</a></li>
diff --git a/objects.inv b/objects.inv
index a3a2de6..1d93679 100644
Binary files a/objects.inv and b/objects.inv differ
diff --git a/searchindex.js b/searchindex.js
index 2f3a6bd..bcae03d 100644
--- a/searchindex.js
+++ b/searchindex.js
@@ -1 +1 @@
-Search.setIndex({"docnames": ["API_reference", "bess", "bibliography", "contrib", "dev_guide", "generated/besskge.batch_sampler", "generated/besskge.batch_sampler.RandomShardedBatchSampler", "generated/besskge.batch_sampler.RigidShardedBatchSampler", "generated/besskge.batch_sampler.ShardedBatchSampler", "generated/besskge.bess", "generated/besskge.bess.AllScoresBESS", "generated/besskge.bess.BessKGE", "generated/besskge.bess.EmbeddingMovingBessKGE", "generated/besskge.bess.ScoreMovingBessKGE", "generated/besskge.bess.TopKQueryBessKGE", "generated/besskge.dataset", "generated/besskge.dataset.KGDataset", "generated/besskge.embedding", "generated/besskge.embedding.init_KGE_normal", "generated/besskge.embedding.init_KGE_uniform", "generated/besskge.embedding.init_uniform_norm", "generated/besskge.embedding.init_xavier_norm", "generated/besskge.embedding.initialize_entity_embedding", "generated/besskge.embedding.initialize_relation_embedding", "generated/besskge.embedding.refactor_embedding_sharding", "generated/besskge.loss", "generated/besskge.loss.BaseLossFunction", "generated/besskge.loss.LogSigmoidLoss", "generated/besskge.loss.MarginBasedLossFunction", "generated/besskge.loss.MarginRankingLoss", "generated/besskge.loss.SampledSoftmaxCrossEntropyLoss", "generated/besskge.metric", "generated/besskge.metric.BaseMetric", "generated/besskge.metric.Evaluation", "generated/besskge.metric.HitsAtK", "generated/besskge.metric.METRICS_DICT", "generated/besskge.metric.ReciprocalRank", "generated/besskge.negative_sampler", "generated/besskge.negative_sampler.PlaceholderNegativeSampler", "generated/besskge.negative_sampler.RandomShardedNegativeSampler", "generated/besskge.negative_sampler.ShardedNegativeSampler", "generated/besskge.negative_sampler.TripleBasedShardedNegativeSampler", "generated/besskge.negative_sampler.TypeBasedShardedNegativeSampler", "generated/besskge.pipeline", "generated/besskge.pipeline.AllScoresPipeline", "generated/besskge.scoring", "generated/besskge.scoring.BaseScoreFunction", "generated/besskge.scoring.BoxE", "generated/besskge.scoring.ComplEx", "generated/besskge.scoring.ConvE", "generated/besskge.scoring.DistMult", "generated/besskge.scoring.DistanceBasedScoreFunction", "generated/besskge.scoring.InterHT", "generated/besskge.scoring.MatrixDecompositionScoreFunction", "generated/besskge.scoring.PairRE", "generated/besskge.scoring.RotatE", "generated/besskge.scoring.TranS", "generated/besskge.scoring.TransE", "generated/besskge.scoring.TripleRE", "generated/besskge.sharding", "generated/besskge.sharding.PartitionedTripleSet", "generated/besskge.sharding.Sharding", "generated/besskge.utils", "generated/besskge.utils.complex_multiplication", "generated/besskge.utils.complex_rotation", "generated/besskge.utils.gather_indices", "generated/besskge.utils.get_entity_filter", "index", "user_guide"], "filenames": ["API_reference.rst", "bess.rst", "bibliography.rst", "contrib.md", "dev_guide.rst", "generated/besskge.batch_sampler.rst", "generated/besskge.batch_sampler.RandomShardedBatchSampler.rst", "generated/besskge.batch_sampler.RigidShardedBatchSampler.rst", "generated/besskge.batch_sampler.ShardedBatchSampler.rst", "generated/besskge.bess.rst", "generated/besskge.bess.AllScoresBESS.rst", "generated/besskge.bess.BessKGE.rst", "generated/besskge.bess.EmbeddingMovingBessKGE.rst", "generated/besskge.bess.ScoreMovingBessKGE.rst", "generated/besskge.bess.TopKQueryBessKGE.rst", "generated/besskge.dataset.rst", "generated/besskge.dataset.KGDataset.rst", "generated/besskge.embedding.rst", "generated/besskge.embedding.init_KGE_normal.rst", "generated/besskge.embedding.init_KGE_uniform.rst", "generated/besskge.embedding.init_uniform_norm.rst", "generated/besskge.embedding.init_xavier_norm.rst", "generated/besskge.embedding.initialize_entity_embedding.rst", "generated/besskge.embedding.initialize_relation_embedding.rst", "generated/besskge.embedding.refactor_embedding_sharding.rst", "generated/besskge.loss.rst", "generated/besskge.loss.BaseLossFunction.rst", "generated/besskge.loss.LogSigmoidLoss.rst", "generated/besskge.loss.MarginBasedLossFunction.rst", "generated/besskge.loss.MarginRankingLoss.rst", "generated/besskge.loss.SampledSoftmaxCrossEntropyLoss.rst", "generated/besskge.metric.rst", "generated/besskge.metric.BaseMetric.rst", "generated/besskge.metric.Evaluation.rst", "generated/besskge.metric.HitsAtK.rst", "generated/besskge.metric.METRICS_DICT.rst", "generated/besskge.metric.ReciprocalRank.rst", "generated/besskge.negative_sampler.rst", "generated/besskge.negative_sampler.PlaceholderNegativeSampler.rst", "generated/besskge.negative_sampler.RandomShardedNegativeSampler.rst", "generated/besskge.negative_sampler.ShardedNegativeSampler.rst", "generated/besskge.negative_sampler.TripleBasedShardedNegativeSampler.rst", "generated/besskge.negative_sampler.TypeBasedShardedNegativeSampler.rst", "generated/besskge.pipeline.rst", "generated/besskge.pipeline.AllScoresPipeline.rst", "generated/besskge.scoring.rst", "generated/besskge.scoring.BaseScoreFunction.rst", "generated/besskge.scoring.BoxE.rst", "generated/besskge.scoring.ComplEx.rst", "generated/besskge.scoring.ConvE.rst", "generated/besskge.scoring.DistMult.rst", "generated/besskge.scoring.DistanceBasedScoreFunction.rst", "generated/besskge.scoring.InterHT.rst", "generated/besskge.scoring.MatrixDecompositionScoreFunction.rst", "generated/besskge.scoring.PairRE.rst", "generated/besskge.scoring.RotatE.rst", "generated/besskge.scoring.TranS.rst", "generated/besskge.scoring.TransE.rst", "generated/besskge.scoring.TripleRE.rst", "generated/besskge.sharding.rst", "generated/besskge.sharding.PartitionedTripleSet.rst", "generated/besskge.sharding.Sharding.rst", "generated/besskge.utils.rst", "generated/besskge.utils.complex_multiplication.rst", "generated/besskge.utils.complex_rotation.rst", "generated/besskge.utils.gather_indices.rst", "generated/besskge.utils.get_entity_filter.rst", "index.rst", "user_guide.rst"], "titles": ["BESS-KGE API Reference", "BESS overview", "Bibliography", "How to contribute to the BESS-KGE project", "How to contribute to the BESS-KGE project", "besskge.batch_sampler", "besskge.batch_sampler.RandomShardedBatchSampler", "besskge.batch_sampler.RigidShardedBatchSampler", "besskge.batch_sampler.ShardedBatchSampler", "besskge.bess", "besskge.bess.AllScoresBESS", "besskge.bess.BessKGE", "besskge.bess.EmbeddingMovingBessKGE", "besskge.bess.ScoreMovingBessKGE", "besskge.bess.TopKQueryBessKGE", "besskge.dataset", "besskge.dataset.KGDataset", "besskge.embedding", "besskge.embedding.init_KGE_normal", "besskge.embedding.init_KGE_uniform", "besskge.embedding.init_uniform_norm", "besskge.embedding.init_xavier_norm", "besskge.embedding.initialize_entity_embedding", "besskge.embedding.initialize_relation_embedding", "besskge.embedding.refactor_embedding_sharding", "besskge.loss", "besskge.loss.BaseLossFunction", "besskge.loss.LogSigmoidLoss", "besskge.loss.MarginBasedLossFunction", "besskge.loss.MarginRankingLoss", "besskge.loss.SampledSoftmaxCrossEntropyLoss", "besskge.metric", "besskge.metric.BaseMetric", "besskge.metric.Evaluation", "besskge.metric.HitsAtK", "besskge.metric.METRICS_DICT", "besskge.metric.ReciprocalRank", "besskge.negative_sampler", "besskge.negative_sampler.PlaceholderNegativeSampler", "besskge.negative_sampler.RandomShardedNegativeSampler", "besskge.negative_sampler.ShardedNegativeSampler", "besskge.negative_sampler.TripleBasedShardedNegativeSampler", "besskge.negative_sampler.TypeBasedShardedNegativeSampler", "besskge.pipeline", "besskge.pipeline.AllScoresPipeline", "besskge.scoring", "besskge.scoring.BaseScoreFunction", "besskge.scoring.BoxE", "besskge.scoring.ComplEx", "besskge.scoring.ConvE", "besskge.scoring.DistMult", "besskge.scoring.DistanceBasedScoreFunction", "besskge.scoring.InterHT", "besskge.scoring.MatrixDecompositionScoreFunction", "besskge.scoring.PairRE", "besskge.scoring.RotatE", "besskge.scoring.TranS", "besskge.scoring.TransE", "besskge.scoring.TripleRE", "besskge.sharding", "besskge.sharding.PartitionedTripleSet", "besskge.sharding.Sharding", "besskge.utils", "besskge.utils.complex_multiplication", "besskge.utils.complex_rotation", "besskge.utils.gather_indices", "besskge.utils.get_entity_filter", "BESS-KGE", "User guide"], "terms": {"when": [1, 3, 4, 11, 12, 13, 14, 23, 26, 27, 28, 29, 30, 37, 38, 41], "distribut": [1, 5, 9, 10, 11, 14, 18, 19, 20, 59, 67, 68], "workload": 1, "over": [1, 6, 7, 8, 14, 67], "n": [1, 41, 60], "worker": [1, 6, 7, 8, 67], "ipu": [1, 3, 4, 6, 7, 8, 9, 10, 44, 65, 67, 68], "randomli": [1, 16], "split": [1, 16, 41], "entiti": [1, 2, 6, 7, 8, 10, 11, 12, 13, 14, 16, 17, 22, 24, 30, 33, 34, 37, 38, 39, 40, 41, 42, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60, 61, 66, 67, 68], "embed": [1, 2, 11, 12, 13, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 67, 68], "tabl": [1, 11, 12, 13, 17, 22, 23, 24, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 68], "shard": [1, 6, 7, 8, 10, 12, 13, 14, 22, 24, 38, 39, 40, 41, 42, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 67], "equal": [1, 12], "size": [1, 6, 7, 8, 10, 13, 14, 41, 44, 47, 48, 49, 50, 52, 54, 55, 56, 57, 58, 68], "each": [1, 5, 6, 7, 8, 10, 11, 12, 13, 14, 16, 20, 22, 23, 33, 39, 41, 44, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60, 61, 66], "which": [1, 12, 16, 38, 39, 40, 41, 42, 66], "i": [1, 3, 4, 6, 7, 8, 12, 13, 14, 16, 22, 23, 34, 39, 41, 44, 47, 49, 60, 64, 65, 66, 67], "store": [1, 13, 15, 16, 44, 60, 67, 68], "": [1, 2, 3, 4], "memori": [1, 6, 7, 8, 11, 12, 13, 67], "The": [1, 3, 4, 6, 7, 8, 10, 11, 12, 13, 16, 24, 26, 27, 28, 29, 30, 33, 41, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60, 61, 66, 67, 68], "relat": [1, 2, 10, 11, 12, 13, 14, 16, 17, 23, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60, 67], "type": [1, 6, 7, 8, 10, 11, 12, 13, 14, 16, 18, 19, 20, 21, 22, 23, 24, 26, 27, 28, 29, 30, 33, 41, 42, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60, 61, 63, 64, 65, 66, 67], "other": [1, 6, 7, 8, 11, 12, 13], "hand": 1, "replic": [1, 13], "across": [1, 3, 4], "usual": 1, "much": 1, "smaller": 1, "figur": 1, "1": [1, 2, 10, 11, 12, 13, 14, 16, 18, 19, 20, 21, 27, 28, 29, 30, 33, 41, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60, 65, 68], "3": [1, 3, 4, 16, 49, 66, 68], "induc": 1, "partit": [1, 6, 7, 8, 10, 11, 14, 38, 39, 40, 41, 42, 44, 60], "tripl": [1, 2, 5, 6, 7, 8, 10, 11, 12, 13, 14, 15, 16, 23, 26, 27, 28, 29, 30, 33, 34, 38, 39, 40, 41, 42, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 66, 67], "dataset": [1, 2, 23, 49, 60, 67, 68], "accord": [1, 5, 11, 12, 13, 18, 19, 20, 21, 24, 33], "pair": [1, 2, 39, 60, 66], "head": [1, 2, 10, 11, 12, 13, 14, 16, 37, 39, 41, 42, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60, 66], "tail": [1, 2, 10, 11, 12, 13, 14, 16, 37, 39, 41, 42, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60, 66], "At": [1, 7, 14], "execut": [1, 59], "time": [1, 6, 7, 8, 67], "both": [1, 3, 4, 26, 46, 67], "train": [1, 9, 11, 12, 13, 16, 43, 44, 67, 68], "infer": [1, 2, 6, 7, 8, 9, 10, 11, 14, 43, 44, 67, 68], "batch": [1, 5, 6, 7, 8, 10, 11, 12, 13, 14, 25, 26, 27, 28, 29, 30, 33, 39, 41, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 63, 64, 67, 68], "ar": [1, 3, 4, 10, 12, 13, 14, 16, 23, 26, 33, 41, 47, 49, 60, 65, 67, 68], "construct": [1, 26, 27, 28, 29, 30, 37, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60, 61, 66], "sampl": [1, 2, 5, 6, 7, 8, 11, 12, 13, 14, 25, 26, 27, 28, 29, 30, 37, 38, 39, 40, 41, 42, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 67, 68], "uniformli": 1, "from": [1, 2, 3, 4, 6, 7, 8, 11, 12, 13, 16, 24, 33, 38, 39, 40, 41, 42, 49, 60, 65], "2": [1, 11, 12, 13, 42, 47, 49, 60, 63, 64, 65, 66, 68], "neg": [1, 5, 6, 7, 8, 10, 11, 12, 13, 14, 16, 25, 26, 27, 28, 29, 30, 37, 38, 39, 40, 41, 42, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60, 68], "us": [1, 2, 3, 4, 6, 7, 8, 10, 11, 13, 14, 16, 26, 27, 28, 29, 30, 33, 37, 38, 39, 41, 44, 49, 51, 60, 67, 68], "corrupt": [1, 6, 7, 8, 10, 11, 12, 13, 33, 37, 38, 39, 40, 41, 42, 49], "order": [1, 14, 16, 33, 41, 44, 60, 68], "also": [1, 3, 4, 61, 68], "balanc": [1, 2, 61, 67], "wai": [1, 16], "ensur": 1, "varieti": 1, "benefici": 1, "final": [1, 3, 4, 14, 47, 49], "qualiti": [1, 16], "left": [1, 3, 4], "A": [1, 2, 33, 38, 60, 61], "made": 1, "up": [1, 3, 4, 68], "9": [1, 68], "block": [1, 10], "contain": [1, 3, 4, 16, 60], "same": [1, 7, 10, 14, 16, 22, 23, 41, 42, 60, 61, 65, 66], "number": [1, 6, 7, 8, 10, 11, 12, 13, 14, 16, 22, 23, 30, 39, 41, 44, 47, 48, 49, 50, 52, 54, 55, 56, 57, 58, 60, 61, 68], "j": [1, 66], "0": [1, 3, 4, 6, 7, 8, 16, 18, 19, 21, 27, 28, 29, 30, 33, 38, 47, 49, 52, 56, 58, 60, 65, 68], "right": 1, "all": [1, 3, 4, 6, 7, 10, 13, 14, 16, 33, 38, 41, 42, 44, 49, 60, 65], "possibli": [1, 16, 41, 44, 49, 60], "pad": [1, 6, 7, 11, 12, 13, 14, 41, 61], "In": [1, 2, 3, 4], "thi": [1, 3, 4, 6, 7, 8, 12, 13, 14, 16, 22, 23, 44, 49, 60, 67], "exampl": [1, 13, 14, 36], "scheme": [1, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 21, 67], "allow": [1, 3, 4, 67], "u": [1, 58], "commun": [1, 67], "first": [1, 3, 4, 16, 39, 66], "need": [1, 16, 22, 23, 26, 27, 28, 29, 30, 44, 47, 49, 67], "gather": [1, 10, 11, 12, 13, 14, 41, 65, 67], "its": [1, 23], "chip": [1, 67], "posit": [1, 5, 6, 7, 8, 11, 12, 13, 19, 25, 26, 27, 28, 29, 30, 33, 45], "These": 1, "includ": [1, 49], "itself": 1, "peer": 1, "requir": [1, 3, 4, 11, 12, 13, 14, 33, 39, 65], "sram": [1, 68], "retriev": [1, 67], "triangl": 1, "colour": 1, "addit": [1, 3, 4, 23], "portion": 1, "can": [1, 3, 4, 12, 13, 14, 22, 23, 33, 44, 60, 67], "reconstruct": 1, "share": [1, 2, 6, 7, 8, 11, 12, 13, 26, 39, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 61, 67], "between": [1, 2, 12, 13, 47, 67], "through": [1, 3, 4, 11, 12, 13, 49], "alltoal": [1, 10, 12, 13, 14], "collect": [1, 6, 7, 8, 10, 11, 12, 13, 14, 15, 60, 67], "oper": [1, 3, 4, 11, 12, 13, 49, 67], "remain": [1, 61], "place": 1, "score": [1, 10, 11, 12, 13, 14, 25, 26, 27, 28, 29, 30, 33, 38, 41, 44, 60, 67], "where": [1, 3, 4, 10, 12, 13, 14, 16, 33, 34, 39, 41, 60], "4": [1, 2, 3, 4, 68], "exchang": 1, "an": [1, 3, 4, 10, 13, 14, 16, 22, 23, 44, 67, 68], "red": 1, "arrow": 1, "effect": [1, 38], "transpos": 1, "row": [1, 20, 33, 41, 63, 64, 65, 66], "column": [1, 16], "pictur": 1, "after": [1, 10, 44, 49, 61], "ha": [1, 6, 7, 8, 20, 68], "correct": [1, 13, 14], "comput": [1, 2, 10, 11, 12, 13, 14, 25, 26, 27, 28, 29, 30, 31, 33, 36, 44, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 67, 68], "present": [1, 16], "abov": [1, 3, 4], "implement": [1, 9], "besskg": [1, 3, 4, 67, 68], "embeddingmovingbesskg": [1, 13, 67], "while": [1, 14], "alwai": [1, 26], "turn": 1, "out": [1, 44], "expens": 1, "mani": 1, "per": [1, 6, 7, 8, 38, 39, 40, 41, 42], "dimens": [1, 33, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "larg": [1, 2, 10, 13, 14, 44], "case": 1, "scoremovingbesskg": [1, 10, 14, 67], "increas": 1, "overal": 1, "throughput": [1, 67], "altern": 1, "work": [1, 3, 4], "well": [1, 3, 4, 52, 56], "differ": [1, 10, 14, 22, 23, 24, 33, 49, 67], "li": [1, 2], "how": 1, "instead": [1, 38, 39, 40, 41, 42, 44, 47], "send": [1, 13], "queri": [1, 10, 13, 14, 33, 38, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60], "devic": [1, 5, 10, 11, 12, 13, 14, 38, 39, 40, 41, 42], "allgath": [1, 13], "against": [1, 10, 12, 13, 14, 38, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60], "partial": 1, "set": [1, 3, 4, 10, 11, 14, 16, 23, 33, 41, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 66, 68], "sent": 1, "via": [1, 2, 10, 14], "new": [1, 3, 4, 6, 7, 8, 24, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 67], "cheaper": 1, "although": 1, "5": 1, "correspond": [1, 13, 22, 23, 41, 65], "6": [1, 2, 3, 4, 47], "put": 1, "back": [1, 13, 61], "came": 1, "complet": [1, 2, 10, 14, 16, 44, 60], "respons": [1, 14], "acls20": [2, 47], "ralph": 2, "abboud": 2, "i\u0307smail": 2, "i\u0307lkan": 2, "ceylan": 2, "thoma": 2, "lukasiewicz": 2, "tommaso": 2, "salvatori": 2, "box": [2, 67], "model": [2, 6, 7, 8, 11, 31, 44, 45, 47, 48, 49, 50, 52, 54, 55, 56, 57, 58, 67], "knowledg": [2, 10, 13, 14, 15, 16, 30, 38, 47, 48, 49, 50, 52, 54, 55, 56, 57, 58, 60, 61, 67, 68], "base": [2, 6, 7, 8, 10, 11, 14, 16, 25, 26, 27, 28, 29, 30, 40, 41, 42, 44, 46, 47, 51, 53, 60], "advanc": 2, "neural": 2, "inform": 2, "process": [2, 5, 6, 7, 8, 38, 39, 40, 41, 42], "system": [2, 3, 4, 68], "33": 2, "annual": 2, "confer": 2, "2020": 2, "neurip": 2, "13": [2, 57], "asam20": [2, 16], "breit": 2, "anna": 2, "ott": 2, "simon": 2, "agibetov": 2, "asan": 2, "samwald": 2, "matthia": 2, "openbiolink": [2, 16], "benchmark": [2, 16], "framework": [2, 11, 67], "scale": [2, 21, 26, 27, 28, 29, 30], "biomed": 2, "link": [2, 3, 4, 33, 67, 68], "predict": [2, 14, 31, 33, 34, 36, 44, 68], "bioinformat": 2, "36": 2, "4097": 2, "4098": 2, "bugd": [2, 57], "antoin": 2, "bord": 2, "nicola": 2, "usuni": 2, "alberto": 2, "garc\u00eda": 2, "dur\u00e1n": 2, "jason": 2, "weston": 2, "oksana": 2, "yakhnenko": 2, "translat": 2, "multi": 2, "data": [2, 6, 7, 8, 10, 16], "26": 2, "27th": 2, "2013": 2, "2787": 2, "2795": 2, "cjm": [2, 9, 10, 11, 14, 30], "22": [2, 9, 10, 11, 14, 30, 52, 58], "cattaneo": 2, "daniel": 2, "justu": 2, "harri": 2, "mellor": 2, "dougla": 2, "orr": 2, "jerom": 2, "maloberti": 2, "zheni": 2, "liu": 2, "thorin": 2, "farnsworth": 2, "andrew": 2, "fitzgibbon": 2, "blazej": 2, "banaszewski": 2, "carlo": 2, "luschi": 2, "bess": [2, 5, 38, 41, 43, 68], "graph": [2, 10, 13, 14, 15, 16, 30, 38, 47, 48, 49, 50, 52, 54, 55, 56, 57, 58, 60, 61, 67, 68], "arxiv": 2, "preprint": 2, "2211": 2, "12281": 2, "2022": 2, "chwc21": [2, 54], "linlin": 2, "chao": 2, "jianshan": 2, "he": 2, "taifeng": 2, "wang": 2, "wei": 2, "chu": 2, "pairr": [2, 67], "vector": 2, "proceed": 2, "59th": 2, "meet": 2, "associ": [2, 16], "linguist": 2, "11th": 2, "intern": [2, 26, 46], "joint": 2, "natur": 2, "languag": 2, "acl": 2, "ijcnlp": 2, "2021": 2, "volum": 2, "long": 2, "paper": 2, "virtual": 2, "event": 2, "august": 2, "4360": 2, "4369": 2, "dppr18": [2, 16, 49], "tim": 2, "dettmer": 2, "minervini": 2, "pasqual": 2, "stenetorp": 2, "pontu": 2, "sebastian": 2, "riedel": 2, "convolut": [2, 49], "2d": [2, 49], "32th": 2, "aaai": 2, "artifici": 2, "intellig": 2, "1811": 2, "1818": 2, "2018": 2, "hfz": [2, 16], "20": [2, 3, 4, 16, 68], "weihua": 2, "hu": 2, "fei": 2, "marinka": 2, "zitnik": 2, "yuxiao": 2, "dong": 2, "hongyu": 2, "ren": 2, "bowen": 2, "michel": 2, "catasta": 2, "jure": 2, "leskovec": 2, "open": [2, 3, 4, 67], "machin": [2, 67], "learn": [2, 23, 47, 48, 49, 50, 52, 54, 55, 56, 57, 58], "jcmb15": [2, 30], "\u00e9": 2, "bastien": 2, "jean": 2, "kyunghyun": 2, "cho": 2, "roland": 2, "memisev": 2, "yoshua": 2, "bengio": 2, "On": 2, "veri": [2, 13], "target": 2, "vocabulari": 2, "53rd": 2, "7th": 2, "10": [2, 16, 44, 68], "2015": 2, "mbs15": [2, 16], "farzaneh": 2, "mahdisoltani": 2, "joanna": 2, "biega": 2, "fabian": 2, "m": [2, 3, 4, 68], "suchanek": 2, "yago3": [2, 16, 68], "multilingu": 2, "wikipedia": 2, "seventh": 2, "biennial": 2, "innov": 2, "research": [2, 3, 4, 68], "cidr": 2, "asilomar": 2, "ca": 2, "usa": 2, "januari": 2, "7": [2, 16, 68], "onlin": 2, "sdnt19": [2, 27, 55], "zhiqe": 2, "sun": 2, "zhi": 2, "hong": 2, "deng": 2, "jian": 2, "yun": 2, "nie": 2, "tang": 2, "rotat": [2, 64, 67], "complex": [2, 55, 63, 64, 67], "space": [2, 47], "represent": [2, 47], "iclr": 2, "2019": 2, "twr": [2, 48], "16": [2, 6, 7, 8, 48, 68], "th": 2, "o": [2, 6, 7, 8], "trouillon": 2, "johann": 2, "welbl": 2, "ric": 2, "gaussier": 2, "guillaum": 2, "bouchard": 2, "simpl": 2, "33rd": 2, "icml": 2, "2016": 2, "48": 2, "jmlr": 2, "workshop": 2, "2071": 2, "2080": 2, "wmw": [2, 52], "baoxin": 2, "qingy": 2, "meng": 2, "ziyu": 2, "honghong": 2, "zhao": 2, "dayong": 2, "wu": 2, "wanxiang": 2, "che": 2, "shijin": 2, "zhigang": 2, "chen": 2, "cong": 2, "interht": [2, 67], "interact": 2, "2202": 2, "04897": 2, "yyh": [2, 50], "15": [2, 16, 50], "bishan": 2, "yang": 2, "wen": 2, "tau": 2, "yih": 2, "xiaodong": 2, "jianfeng": 2, "gao": 2, "3rd": 2, "yll": [2, 58], "yu": 2, "zhicong": 2, "luo": 2, "huanyong": 2, "lin": 2, "hongzhu": 2, "yafeng": 2, "tripler": [2, 67], "2209": 2, "08271": 2, "zyx22": [2, 56], "xuanyu": 2, "zhang": 2, "qing": 2, "dongliang": 2, "xu": 2, "tran": [2, 67], "transit": 2, "synthet": 2, "find": [2, 66], "emnlp": 2, "1202": 2, "1208": 2, "you": [3, 4, 68], "even": [3, 4], "don": [3, 4], "t": [3, 4, 10, 12, 14, 15, 16, 38, 39, 40, 41, 42, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60, 66, 67], "have": [3, 4, 16, 44, 49, 67], "access": [3, 4, 67], "ipumodel": [3, 4], "emul": [3, 4], "most": [3, 4, 14, 34, 44], "function": [3, 4, 6, 7, 8, 10, 11, 12, 13, 14, 17, 22, 23, 25, 26, 27, 28, 29, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 62, 65, 68], "physic": [3, 4, 68], "hardwar": [3, 4, 44], "tunnel": [3, 4], "web": [3, 4], "editor": [3, 4], "desktop": [3, 4], "app": [3, 4], "minimum": [3, 4], "effort": [3, 4], "excel": [3, 4], "solut": [3, 4], "test": [3, 4, 16, 68], "directli": [3, 4], "here": [3, 4, 16, 44], "do": [3, 4], "fork": [3, 4], "repositori": [3, 4], "launch": [3, 4], "hour": [3, 4], "session": [3, 4], "free": [3, 4, 67, 68], "form": [3, 4], "http": [3, 4, 16, 68], "consol": [3, 4], "com": [3, 4, 16, 68], "github": [3, 4, 16, 67, 68], "userid": [3, 4], "reponam": [3, 4], "graphcor": [3, 4, 68], "2fpytorch": [3, 4], "3a3": [3, 4], "ubuntu": [3, 4, 68], "04": [3, 4, 68], "20230703": [3, 4], "pod4": [3, 4, 68], "repopnam": [3, 4], "address": [3, 4], "e": [3, 4, 16, 63, 64, 65, 66], "g": [3, 4], "origin": [3, 4, 16], "repo": [3, 4, 67], "start": [3, 4, 33, 67], "clone": [3, 4], "termin": [3, 4], "pane": [3, 4], "run": [3, 4, 6, 7, 8, 44, 68], "command": [3, 4], "bash": [3, 4], "gradient": [3, 4, 68], "launch_vscode_serv": [3, 4], "sh": [3, 4, 68], "name": [3, 4, 10, 12, 14, 16, 44, 60], "option": [3, 4, 6, 7, 8, 10, 11, 12, 13, 14, 16, 22, 23, 33, 41, 44, 60, 61, 68], "argument": [3, 4, 49], "defin": [3, 4, 16, 47], "remot": [3, 4], "default": [3, 4, 6, 7, 8, 10, 11, 12, 13, 14, 18, 19, 21, 29, 33, 39, 41, 44, 47, 48, 49, 50, 52, 54, 55, 56, 57, 58, 60, 61], "script": [3, 4], "download": [3, 4, 16], "instal": [3, 4, 67], "depend": [3, 4, 66, 68], "ask": [3, 4], "author": [3, 4], "account": [3, 4], "write": [3, 4], "privileg": [3, 4], "provid": [3, 4, 14, 22, 23, 33, 44], "pleas": [3, 4], "refer": [3, 4, 67], "notebook": [3, 4, 68], "detail": [3, 4, 68], "step": [3, 4, 10, 11, 12, 13, 14, 44], "connect": [3, 4], "onc": [3, 4, 22, 23, 67], "dev": [3, 4], "build": [3, 4, 15, 16], "custom": [3, 4], "op": [3, 4], "now": [3, 4], "readi": [3, 4], "close": [3, 4], "stop": [3, 4], "rememb": [3, 4], "unregist": [3, 4], "explain": [3, 4], "common": [3, 4], "issu": [3, 4, 67], "paragraph": [3, 4], "To": [3, 4, 10, 11, 14, 44], "resum": [3, 4], "your": [3, 4, 68], "just": [3, 4], "section": [3, 4], "profil": [3, 4], "repeat": [3, 4, 7], "chang": [3, 4, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "extens": [3, 4], "persist": [3, 4], "poplar": [3, 4, 68], "sdk": [3, 4, 68], "follow": [3, 4, 68], "instruct": [3, 4, 68], "get": [3, 4, 67], "guid": [3, 4, 67], "Then": [3, 4], "enabl": [3, 4, 68], "creat": [3, 4, 60, 61, 68], "activ": [3, 4, 29, 67, 68], "python": [3, 4, 67, 68], "virtualenv": [3, 4, 68], "poptorch": [3, 4, 6, 7, 8, 67, 68], "wheel": [3, 4, 68], "necessari": [3, 4], "python3": [3, 4, 68], "8": [3, 4, 68], "venv": [3, 4, 68], "add": [3, 4], "bin": [3, 4, 68], "sourc": [3, 4, 6, 7, 8, 10, 11, 12, 13, 14, 16, 18, 19, 20, 21, 22, 23, 24, 26, 27, 28, 29, 30, 32, 33, 34, 36, 38, 39, 40, 41, 42, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60, 61, 63, 64, 65, 66, 68], "path_to_poplar_sdk": [3, 4], "pip": [3, 4, 68], "poplar_sdk_en": [3, 4, 68], "whl": [3, 4, 68], "r": [3, 4, 10, 14, 15, 16, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60, 64, 66, 67], "txt": [3, 4], "sever": [3, 4], "util": [3, 4, 6, 7, 8, 15, 17, 31, 67], "dure": [3, 4, 68], "check": [3, 4], "help": [3, 4], "list": [3, 4, 6, 7, 8, 16, 22, 23, 33, 41, 44, 47, 48, 49, 50, 52, 54, 55, 56, 57, 58], "befor": [3, 4, 11, 12, 13, 14, 47, 49, 52, 54, 56, 58], "submit": [3, 4], "pr": [3, 4], "upstream": [3, 4], "ci": [3, 4], "particular": [3, 4], "mind": [3, 4], "our": [3, 4, 68], "format": [3, 4], "error": [3, 4, 10, 14, 44, 68], "lint": [3, 4], "automat": [3, 4, 16], "insid": [3, 4, 47, 68], "unit": [3, 4], "folder": [3, 4], "individu": [3, 4], "pattern": [3, 4], "match": [3, 4], "filter": [3, 4, 11, 12, 13, 14, 44, 66], "k": [3, 4, 14, 33, 34, 35, 44, 64, 65], "cpp": [3, 4], "custom_op": [3, 4], "updat": [3, 4], "makefil": [3, 4], "ad": [3, 4, 23, 60, 67], "class": [5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61], "partitioned_triple_set": [6, 7, 8, 44], "negative_sampl": [6, 7, 8, 10, 11, 12, 13, 14, 67], "shard_b": [6, 7, 8, 10, 14], "batches_per_step": [6, 7, 8], "seed": [6, 7, 8, 16, 38, 39, 41, 42, 61], "hrt_freq_weight": [6, 7, 8], "fals": [6, 7, 8, 11, 12, 13, 14, 33, 39, 41, 44, 47, 48, 50, 52, 54, 55, 56, 57, 58, 60], "weight_smooth": [6, 7, 8], "duplicate_batch": [6, 7, 8], "return_triple_idx": [6, 7, 8], "random": [6, 16, 39, 61], "indic": [6, 7, 8, 10, 11, 12, 13, 14, 33, 41, 60, 65], "replac": 6, "No": [6, 38, 68], "appli": [6, 41, 44, 49, 52, 56, 61, 66], "initi": [6, 7, 8, 10, 11, 12, 13, 14, 17, 18, 19, 20, 21, 22, 23, 26, 27, 28, 29, 30, 33, 38, 39, 41, 42, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "sampler": [6, 7, 8, 10, 11, 12, 13, 14, 38, 39, 40, 41, 42, 44, 49], "paramet": [6, 7, 8, 10, 11, 12, 13, 14, 16, 18, 19, 20, 21, 22, 23, 24, 26, 27, 28, 29, 30, 33, 34, 36, 38, 39, 41, 42, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60, 61, 63, 64, 65, 66, 68], "partitionedtripleset": [6, 7, 8, 49, 67], "pre": [6, 7, 8, 16], "shardednegativesampl": [6, 7, 8, 11, 12, 13, 38, 67], "int": [6, 7, 8, 10, 11, 12, 13, 14, 16, 22, 23, 30, 34, 38, 39, 41, 42, 44, 47, 48, 49, 50, 51, 52, 54, 55, 56, 57, 58, 61], "micro": [6, 7, 8, 11, 12, 13, 14], "call": [6, 7, 8, 38], "rng": [6, 7, 8, 38, 39, 40, 41, 42], "bool": [6, 7, 8, 11, 12, 13, 14, 18, 19, 23, 26, 27, 28, 29, 30, 33, 38, 39, 40, 41, 42, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60], "If": [6, 7, 8, 11, 12, 13, 14, 16, 22, 23, 33, 39, 41, 44, 47, 48, 49, 50, 52, 54, 55, 56, 57, 58, 60, 65, 68], "true": [6, 7, 8, 11, 12, 13, 14, 18, 19, 23, 33, 39, 41, 44, 47, 48, 49, 50, 52, 54, 55, 56, 57, 58, 61], "frequenc": [6, 7, 8], "weight": [6, 7, 8, 11, 12, 13, 26, 27, 28, 29, 30, 68], "float": [6, 7, 8, 16, 18, 19, 21, 27, 28, 29, 30, 47, 49, 52, 56, 58], "smooth": [6, 7, 8], "two": [6, 7, 8, 47, 49, 66], "ident": [6, 7, 8], "halv": [6, 7, 8], "ht": [6, 7, 8, 12, 38, 39, 40, 41, 42], "return": [6, 7, 8, 10, 11, 12, 13, 14, 16, 18, 19, 20, 21, 22, 23, 24, 26, 27, 28, 29, 30, 33, 34, 36, 38, 41, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60, 61, 63, 64, 65, 66], "wrt": [6, 7, 8, 16, 44], "get_dataload": [6, 7, 8], "shuffl": [6, 7, 8], "num_work": [6, 7, 8], "persistent_work": [6, 7, 8], "buffer_s": [6, 7, 8], "dataload": [6, 7, 8], "instanti": [6, 7, 8, 16], "appropri": [6, 7, 8], "iter": [6, 7, 8, 10, 14], "It": [6, 7, 8, 14, 44], "asynchron": [6, 7, 8], "load": [6, 7, 8, 14, 16, 61], "minim": [6, 7, 8, 67], "cpu": [6, 7, 8], "compil": [6, 7, 8, 68], "epoch": [6, 7, 8], "see": [6, 7, 8, 10, 11, 12, 13, 14, 27, 28, 29, 30, 33, 36, 38, 41, 42, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 67, 68], "torch": [6, 7, 8, 65], "__init__": [6, 7, 8, 29, 41, 42, 47, 48, 49, 50, 52, 54, 55, 56, 57, 58], "ring": [6, 7, 8], "buffer": [6, 7, 8], "preload": [6, 7, 8], "get_dataloader_sampl": [6, 7, 8], "sample_tripl": [6, 7, 8], "idx": [6, 7, 8], "index": [6, 7, 8, 10, 41, 65, 66], "dict": [6, 7, 8, 11, 12, 13, 14, 16, 33, 44], "str": [6, 7, 8, 11, 12, 13, 14, 16, 29, 33, 35, 38, 39, 40, 41, 42, 44, 60, 66], "union": [6, 7, 8, 14, 16, 22, 23, 44, 47, 48, 49, 50, 52, 54, 55, 56, 57, 58], "ndarrai": [6, 7, 8, 16, 41, 42, 44, 60, 61], "ani": [6, 7, 8, 11, 12, 13, 14, 16, 41, 42, 44, 60, 61, 67], "dtype": [6, 7, 8, 16, 41, 42, 44, 60, 61, 68], "int64": [6, 7, 8, 41, 60, 61], "bool_": [6, 7, 8, 41], "relev": [6, 7, 8, 11, 12, 13], "static": [6, 7, 8], "worker_init_fn": [6, 7, 8], "worker_id": [6, 7, 8], "pass": [6, 7, 8, 14, 22, 23, 24, 44, 47, 49], "id": [6, 7, 8, 16, 23, 33, 41, 42, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60, 61, 66], "none": [6, 7, 8, 10, 11, 12, 13, 14, 16, 22, 23, 33, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60, 61], "specifi": [7, 10, 16, 44, 65], "shorter": 7, "ones": [7, 23], "length": [7, 22, 23, 41], "mask": [7, 11, 12, 13, 14, 33, 41], "identifi": [7, 11, 12, 13], "abstract": [8, 11, 26, 28, 46, 51, 53], "pytorch": [9, 68], "modul": [9, 10, 11, 12, 13, 14, 26, 31, 33, 44, 46], "kge": [9, 10, 11, 12, 13, 14, 31, 45, 68], "multipl": [9, 63], "candidate_sampl": [10, 14], "score_fn": [10, 11, 12, 13, 14, 44], "window_s": [10, 14, 44], "1000": [10, 44], "h": [10, 12, 14, 15, 16, 38, 39, 40, 41, 42, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60, 66, 67], "host": [10, 11, 12, 13, 44], "combin": [10, 11, 14, 44, 49], "h_shard": [10, 14, 44, 60], "t_shard": [10, 14, 44, 60], "sinc": 10, "onli": [10, 14, 16, 38, 39, 40, 41, 42, 49, 60], "part": [10, 16, 60, 63, 64], "slide": [10, 14, 44], "window": [10, 14, 44], "metric": [10, 11, 12, 13, 14, 44, 67], "should": [10, 14, 16, 44, 49], "aggreg": 10, "pipelin": [10, 67], "allscorespipelin": [10, 67], "allscor": 10, "placeholdernegativesampl": [10, 14, 67], "basescorefunct": [10, 11, 12, 13, 14, 44, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 67], "decreas": [10, 14, 33, 44], "avoid": [10, 13, 14, 44], "oom": [10, 14, 44], "forward": [10, 11, 12, 13, 14, 26, 27, 28, 29, 30, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "similarli": [10, 14, 60], "candid": [10, 14, 33, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "thei": [10, 13, 14, 41], "togeth": [10, 14], "tensor": [10, 11, 12, 13, 14, 18, 19, 20, 21, 22, 23, 26, 27, 28, 29, 30, 33, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 63, 64, 65, 66], "self": [10, 26, 27, 28, 29, 30], "shape": [10, 11, 12, 13, 14, 16, 22, 24, 26, 27, 28, 29, 30, 33, 41, 42, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60, 61, 63, 64, 65, 66], "known": [10, 14, 60], "loss_fn": [11, 12, 13], "evalu": [11, 12, 13, 14, 31, 44, 67], "return_scor": [11, 12, 13, 14, 44], "augment_neg": [11, 12, 13], "ht_shardpair": [11, 60], "baselossfunct": [11, 12, 13, 27, 28, 29, 30, 67], "loss": [11, 12, 13, 49, 67], "augment": [11, 12, 13], "triple_mask": [11, 12, 13, 14, 33], "triple_weight": [11, 12, 13, 26, 27, 28, 29, 30], "negative_mask": [11, 12, 13, 14], "compris": [11, 12, 13], "four": [11, 12, 13], "phase": [11, 12, 13], "local": [11, 12, 13, 16, 60, 61, 67], "n_shard": [11, 12, 13, 14, 22, 41, 60, 61], "positive_per_partit": [11, 12, 13], "b": [11, 12, 13, 14, 19, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 65], "padded_neg": [11, 12, 13, 14, 41], "discard": [11, 12, 13, 14], "properti": [11, 12, 13, 16, 61], "n_embedding_paramet": [11, 12, 13], "trainabl": [11, 12, 13], "score_batch": [11, 12, 13], "tupl": [11, 12, 13, 16, 41, 66], "n_neg": [11, 12, 13, 26, 27, 28, 29, 30, 33, 39, 41, 42, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 60], "move": [12, 13], "done": 12, "singl": [12, 33], "total": [12, 30], "disabl": 12, "otherwis": [12, 65], "conveni": 13, "so": [13, 20], "For": [13, 14, 44, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 65, 67, 68], "valu": 13, "document": 13, "multipli": [13, 52, 56], "doe": 13, "support": [13, 19, 33, 44, 68], "100": [14, 68], "specif": [14, 16, 41, 44, 45, 60], "top": [14, 33, 44], "like": [14, 33, 34, 44, 65], "input": [14, 49], "recommend": [14, 68], "one": [14, 23, 24, 41, 60], "want": [14, 16], "loop": 14, "topk": 14, "triplebasedshardednegativesampl": [14, 67], "unnecessari": 14, "best": 14, "respect": 14, "kept": 14, "next": 14, "rest": 14, "mask_on_gath": [14, 41], "n_entiti": [16, 22, 30, 61], "n_relation_typ": [16, 23, 47, 48, 49, 50, 52, 54, 55, 56, 57, 58], "original_triple_id": 16, "entity_dict": 16, "relation_dict": 16, "type_offset": [16, 61], "neg_head": [16, 60], "neg_tail": [16, 60], "repres": 16, "int32": [16, 41, 42, 44, 60, 61], "classmethod": [16, 60, 61], "build_ogbl_biokg": 16, "root": 16, "ogbl": [16, 68], "biokg": [16, 68], "ogb": 16, "stanford": 16, "edu": 16, "doc": 16, "linkprop": 16, "path": [16, 61, 68], "locat": 16, "build_ogbl_wikikg2": 16, "wikikg2": [16, 68], "build_openbiolink": 16, "high": [16, 43, 67], "version": 16, "openbiolink2020": 16, "hq": 16, "build_yago310": 16, "subgraph": 16, "least": 16, "them": [16, 41, 67, 68], "yago": 16, "org": 16, "label": 16, "from_datafram": 16, "df": 16, "head_column": 16, "relation_column": 16, "tail_column": 16, "entity_typ": 16, "1234": 16, "panda": 16, "datafram": 16, "assign": [16, 33, 60], "contigu": 16, "dictionari": [16, 33], "seri": 16, "map": [16, 61], "string": 16, "valid": 16, "instanc": 16, "from_tripl": 16, "arrai": 16, "alreadi": [16, 41], "been": [16, 49, 67, 68], "note": [16, 49], "attribut": [16, 31], "manual": 16, "numpi": 16, "head_id": 16, "relation_id": [16, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "tail_id": 16, "num_tripl": 16, "offset": [16, 52, 56, 58, 61], "ht_type": 16, "n_tripl": [16, 41, 42, 60], "h_type": [16, 60], "t_type": [16, 60], "object": [16, 61], "save": [16, 61, 68], "node": 16, "edg": 16, "n_neg_head": [16, 60], "n_neg_tail": [16, 60], "out_fil": [16, 61], "pkl": 16, "output": [16, 33, 49, 61], "file": [16, 61, 68], "h_id": 16, "r_id": 16, "t_id": 16, "assum": [16, 33, 49, 68], "cluster": [16, 41, 61], "manag": 17, "embedding_t": [18, 19, 20, 21], "std": 18, "divide_by_embedding_s": [18, 19], "normal": [18, 20, 21, 47, 49, 52, 54, 56, 58], "mean": 18, "standard": [18, 21], "deviat": [18, 21], "rescal": [18, 19], "row_siz": [18, 19, 21, 22, 23, 24], "symmetr": 19, "uniform": [19, 20], "boundari": 19, "norm": [20, 47, 51, 52, 54, 55, 56, 57, 58], "gain": 21, "xavier": 21, "fan_in": 21, "fan_out": 21, "factor": [21, 26, 27, 28, 29, 30, 58], "callabl": [22, 23, 35, 47, 48, 49, 50, 52, 54, 55, 56, 57, 58], "either": 22, "max_entity_per_shard": [22, 61], "unshard": 22, "alloc": [22, 23], "entri": [22, 23], "omit": [22, 23], "max_ent_per_shard": 22, "inverse_rel": [23, 47, 48, 49, 50, 52, 54, 55, 56, 57, 58], "invers": [23, 47, 48, 49, 50, 52, 54, 55, 56, 57, 58, 60], "direct": 23, "given": [23, 44, 66, 68], "entity_embed": [24, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "old_shard": 24, "new_shard": [24, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "refactor": 24, "n_shard_old": 24, "max_ent_per_shard_old": 24, "current": [24, 33], "n_shard_new": 24, "max_ent_per_shard_new": 24, "arg": [26, 46], "kwarg": [26, 46], "fp32": [26, 68], "state": [26, 46], "nn": [26, 46], "scriptmodul": [26, 46], "positive_scor": [26, 27, 28, 29, 30], "negative_scor": [26, 27, 28, 29, 30], "batch_siz": [26, 27, 28, 29, 30, 33, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "get_negative_weight": [26, 27, 28, 29, 30], "negative_adversarial_sampl": [26, 27, 28, 29, 30], "els": [26, 27, 28, 29, 30, 33, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "loss_scal": [26, 27, 28, 29, 30], "might": [26, 27, 28, 29, 30], "fp16": [26, 27, 28, 29, 30, 68], "adversari": [26, 27, 28, 29, 30], "negative_adversarial_scal": [26, 27, 28, 29, 30], "reciproc": [26, 27, 28, 29, 30, 36], "temperatur": [26, 27, 28, 29, 30], "margin": [27, 28, 29], "log": 27, "sigmoid": [27, 49], "activation_funct": 29, "relu": 29, "rank": [29, 33, 34, 36], "pairwis": 29, "hing": 29, "marginbasedlossfunct": [29, 67], "softmax": 30, "cross": 30, "entropi": 30, "metric_list": 33, "mode": [33, 60], "averag": 33, "worst_rank_infti": 33, "reduct": [33, 36, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "return_rank": 33, "mrr": [33, 35, 36], "hit": [33, 34, 35], "optimist": 33, "pessimist": 33, "infin": 33, "worst": 33, "possibl": 33, "method": 33, "reduc": [33, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "along": [33, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 65], "sum": [33, 47, 48, 49, 50, 53], "alongsid": 33, "dict_metrics_from_rank": 33, "batch_rank": 33, "element": [33, 65], "boolean": 33, "ranks_from_indic": 33, "ground_truth": [33, 60], "candidate_indic": 33, "ground": [33, 34, 36, 60], "truth": [33, 34, 36, 60], "n_candid": 33, "likelihood": 33, "distinct": 33, "among": [33, 34, 36], "ranks_from_scor": 33, "pos_scor": 33, "candidate_scor": 33, "stacked_metrics_from_rank": 33, "stack": 33, "n_metric": 33, "count": [34, 68], "maximum": [34, 68], "accept": 34, "hitsatk": [35, 67], "reciprocalrank": [35, 67], "basemetr": [36, 67], "corruption_schem": [38, 39, 40, 41, 42, 44], "placehold": 38, "topkquerybesskg": [38, 41, 67], "flat_negative_format": [38, 39, 40, 41, 42], "local_sampl": [38, 39, 40, 41, 42], "gener": [38, 39, 40, 41, 42, 62], "half": 39, "second": [39, 66, 68], "negative_head": 41, "negative_tail": 41, "return_sort_idx": 41, "predetermin": 41, "global": [41, 44, 60, 61, 66], "randomshardednegativesampl": [41, 42, 67], "sort": [41, 60], "recov": 41, "pad_neg": 41, "shard_count": [41, 61], "padded_shard_length": 41, "divid": 41, "view": 41, "shard_neg": 41, "shard_neg_count": 41, "sort_neg_idx": 41, "triple_typ": 42, "level": 43, "api": [43, 67, 68], "batch_sampl": [44, 67], "filter_tripl": [44, 66], "return_topk": 44, "use_ipu_model": 44, "kg": 44, "appear": [44, 66], "shardedbatchsampl": [44, 67], "whose": 44, "must": 44, "caus": 44, "go": 44, "actual": 44, "result": 44, "head_emb": [46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "tail_emb": [46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "score_tripl": [46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "negative_sample_shar": [46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "relation_embed": [46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "score_head": [46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "fix": [46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "n_head": [46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "embedding_s": [46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "broadcast": [46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "score_tail": [46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "n_tail": [46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "update_shard": [46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "scoring_norm": [47, 51, 52, 54, 55, 56, 57, 58], "entity_initi": [47, 48, 49, 50, 52, 54, 55, 56, 57, 58], "uniform_": 47, "relation_initi": [47, 48, 49, 50, 52, 54, 55, 56, 57, 58], "init_uniform_norm": [47, 67], "apply_tanh": 47, "dist_func_per_dim": 47, "ep": 47, "1e": 47, "06": 47, "distancebasedscorefunct": [47, 48, 49, 50, 52, 54, 55, 56, 57, 58, 67], "center": 47, "scalar": [47, 49], "bound": [47, 67], "bump": 47, "tanh": 47, "select": 47, "distanc": [47, 51, 52, 54, 55, 56, 57, 58], "whether": [47, 60], "outsid": 47, "make": [47, 60], "choic": 47, "separ": 47, "soften": 47, "geometr": 47, "width": [47, 49], "boxe_scor": 47, "bumped_ht": 47, "center_ht": 47, "width_ht": 47, "box_siz": 47, "optim": [47, 68], "emb_siz": 47, "control": 47, "broadcasted_dist": [47, 51, 52, 54, 55, 56, 57, 58], "v1": [47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 63], "v2": [47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 63], "p": [47, 51, 52, 54, 55, 56, 57, 58], "reduce_embed": [47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "v": [47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 64, 67], "init_kge_norm": [48, 67], "broadcasted_dot_product": [48, 49, 50, 53], "dot": [48, 49, 50, 53], "product": [48, 49, 50, 53], "embedding_height": 49, "embedding_width": 49, "init_xavier_norm": [49, 67], "zeros_": 49, "input_channel": 49, "output_channel": 49, "32": 49, "kernel_height": 49, "kernel_width": 49, "input_dropout": 49, "feature_map_dropout": 49, "hidden_dropout": 49, "batch_norm": 49, "layer": 49, "we": [49, 68], "By": 49, "design": [49, 67], "add_inverse_tripl": [49, 60], "create_from_dataset": [49, 60], "height": 49, "reshap": 49, "concaten": 49, "bias": 49, "channel": 49, "conv2d": 49, "kernel": 49, "rate": 49, "dropout": 49, "linear": 49, "init_kge_uniform": [50, 52, 54, 55, 56, 57, 58, 67], "normalize_ent": [52, 54, 56, 58], "l2": [52, 54, 56, 58], "auxiliari": 52, "matrix": 53, "decomposit": 53, "project": [54, 58], "real": [55, 63, 64], "tild": 56, "triplerev2": 58, "inverse_tripl": 60, "partition_mod": 60, "dummi": 60, "triple_count": 60, "triple_offset": 60, "triple_sort_idx": 60, "shard_h": 60, "shard_t": 60, "kgdataset": [60, 67], "create_from_queri": 60, "query_mod": 60, "negative_typ": 60, "n_queri": 60, "hr": 60, "rt": 60, "resp": 60, "r_inv": 60, "regular": 60, "criterion": 60, "delimit": 60, "entity_to_shard": 61, "entity_to_idx": 61, "shard_and_idx_to_ent": 61, "entity_type_count": 61, "entity_type_offset": 61, "again": 61, "n_type": 61, "npz": 61, "local_id": 61, "exclud": 61, "purpos": 62, "imaginari": [63, 64], "wise": [63, 64], "unitari": 64, "pi": 64, "x": [65, 66], "friendli": 65, "take_along_dim": 65, "dimension": 65, "dim": 65, "take": 65, "filter_mod": 66, "compar": 66, "y": 66, "determin": 66, "look": 66, "z": 66, "spars": 66, "packag": 67, "shallow": 67, "typic": 67, "littl": 67, "perform": 67, "maxim": 67, "bandwidth": 67, "fast": 67, "leverag": 67, "achiev": 67, "introduct": 67, "overview": 67, "librari": [67, 68], "still": 67, "develop": 67, "featur": 67, "expect": 67, "overtim": 67, "occasion": 67, "bug": 67, "mai": 67, "occur": 67, "feel": 67, "report": 67, "problem": 67, "user": 67, "usag": 67, "limit": 67, "randomshardedbatchsampl": 67, "rigidshardedbatchsampl": 67, "typebasedshardednegativesampl": 67, "allscoresbess": 67, "conv": 67, "distmult": 67, "matrixdecompositionscorefunct": 67, "trans": 67, "logsigmoidloss": 67, "marginrankingloss": 67, "sampledsoftmaxcrossentropyloss": 67, "metrics_dict": 67, "initialize_entity_embed": 67, "initialize_relation_embed": 67, "refactor_embedding_shard": 67, "complex_multipl": 67, "complex_rot": 67, "gather_indic": 67, "get_entity_filt": 67, "code": 67, "server": 67, "paperspac": [67, 68], "setup": 67, "tip": 67, "bibliographi": 67, "popart": 68, "more": 68, "quick": 68, "git": 68, "import": 68, "1403": 68, "walkthrough": 68, "main": 68, "jupyt": 68, "sequenc": 68, "click": 68, "button": 68, "avail": 68, "introduc": 68, "therefor": 68, "some": 68, "approxim": 68, "estim": 68, "below": 68, "accumul": 68, "momentum": 68, "notic": 68, "cap": 68, "max": 68, "pod16": 68, "float16": 68, "sgdm": 68, "2m": 68, "2e8": 68, "13m": 68, "3e9": 68, "128": 68, "adam": 68, "4m": 68, "0e8": 68, "9m": 68, "256": 68, "ye": 68, "900k": 68, "3e8": 68, "5m": 68, "8m": 68, "2e9": 68, "512": 68, "375k": 68, "9e8": 68, "7e8": 68, "messag": 68, "about": 68, "onnx": 68, "protobuff": 68, "exceed": 68, "_popart": 68, "saveinitializerstofil": 68, "my_fil": 68}, "objects": {"": [[67, 0, 0, "-", "besskge"]], "besskge": [[5, 0, 0, "-", "batch_sampler"], [9, 0, 0, "-", "bess"], [15, 0, 0, "-", "dataset"], [17, 0, 0, "-", "embedding"], [25, 0, 0, "-", "loss"], [31, 0, 0, "-", "metric"], [37, 0, 0, "-", "negative_sampler"], [43, 0, 0, "-", "pipeline"], [45, 0, 0, "-", "scoring"], [59, 0, 0, "-", "sharding"], [62, 0, 0, "-", "utils"]], "besskge.batch_sampler": [[6, 1, 1, "", "RandomShardedBatchSampler"], [7, 1, 1, "", "RigidShardedBatchSampler"], [8, 1, 1, "", "ShardedBatchSampler"]], "besskge.batch_sampler.RandomShardedBatchSampler": [[6, 2, 1, "", "get_dataloader"], [6, 2, 1, "", "get_dataloader_sampler"], [6, 2, 1, "", "sample_triples"], [6, 2, 1, "", "worker_init_fn"]], "besskge.batch_sampler.RigidShardedBatchSampler": [[7, 2, 1, "", "get_dataloader"], [7, 2, 1, "", "get_dataloader_sampler"], [7, 2, 1, "", "sample_triples"], [7, 2, 1, "", "worker_init_fn"]], "besskge.batch_sampler.ShardedBatchSampler": [[8, 2, 1, "", "get_dataloader"], [8, 2, 1, "", "get_dataloader_sampler"], [8, 2, 1, "", "sample_triples"], [8, 2, 1, "", "worker_init_fn"]], "besskge.bess": [[10, 1, 1, "", "AllScoresBESS"], [11, 1, 1, "", "BessKGE"], [12, 1, 1, "", "EmbeddingMovingBessKGE"], [13, 1, 1, "", "ScoreMovingBessKGE"], [14, 1, 1, "", "TopKQueryBessKGE"]], "besskge.bess.AllScoresBESS": [[10, 2, 1, "", "forward"]], "besskge.bess.BessKGE": [[11, 2, 1, "", "forward"], [11, 3, 1, "", "n_embedding_parameters"], [11, 2, 1, "", "score_batch"]], "besskge.bess.EmbeddingMovingBessKGE": [[12, 2, 1, "", "forward"], [12, 3, 1, "", "n_embedding_parameters"], [12, 2, 1, "", "score_batch"]], "besskge.bess.ScoreMovingBessKGE": [[13, 2, 1, "", "forward"], [13, 3, 1, "", "n_embedding_parameters"], [13, 2, 1, "", "score_batch"]], "besskge.bess.TopKQueryBessKGE": [[14, 2, 1, "", "forward"]], "besskge.dataset": [[16, 1, 1, "", "KGDataset"]], "besskge.dataset.KGDataset": [[16, 2, 1, "", "build_ogbl_biokg"], [16, 2, 1, "", "build_ogbl_wikikg2"], [16, 2, 1, "", "build_openbiolink"], [16, 2, 1, "", "build_yago310"], [16, 4, 1, "", "entity_dict"], [16, 2, 1, "", "from_dataframe"], [16, 2, 1, "", "from_triples"], [16, 3, 1, "", "ht_types"], [16, 2, 1, "", "load"], [16, 4, 1, "", "n_entity"], [16, 4, 1, "", "n_relation_type"], [16, 4, 1, "", "neg_heads"], [16, 4, 1, "", "neg_tails"], [16, 4, 1, "", "original_triple_ids"], [16, 4, 1, "", "relation_dict"], [16, 2, 1, "", "save"], [16, 4, 1, "", "triples"], [16, 4, 1, "", "type_offsets"]], "besskge.embedding": [[18, 5, 1, "", "init_KGE_normal"], [19, 5, 1, "", "init_KGE_uniform"], [20, 5, 1, "", "init_uniform_norm"], [21, 5, 1, "", "init_xavier_norm"], [22, 5, 1, "", "initialize_entity_embedding"], [23, 5, 1, "", "initialize_relation_embedding"], [24, 5, 1, "", "refactor_embedding_sharding"]], "besskge.loss": [[26, 1, 1, "", "BaseLossFunction"], [27, 1, 1, "", "LogSigmoidLoss"], [28, 1, 1, "", "MarginBasedLossFunction"], [29, 1, 1, "", "MarginRankingLoss"], [30, 1, 1, "", "SampledSoftmaxCrossEntropyLoss"]], "besskge.loss.BaseLossFunction": [[26, 2, 1, "", "forward"], [26, 2, 1, "", "get_negative_weights"], [26, 4, 1, "", "loss_scale"], [26, 4, 1, "", "negative_adversarial_sampling"], [26, 4, 1, "", "negative_adversarial_scale"]], "besskge.loss.LogSigmoidLoss": [[27, 2, 1, "", "forward"], [27, 2, 1, "", "get_negative_weights"], [27, 4, 1, "", "loss_scale"], [27, 4, 1, "", "negative_adversarial_sampling"], [27, 4, 1, "", "negative_adversarial_scale"]], "besskge.loss.MarginBasedLossFunction": [[28, 2, 1, "", "forward"], [28, 2, 1, "", "get_negative_weights"], [28, 4, 1, "", "loss_scale"], [28, 4, 1, "", "negative_adversarial_sampling"], [28, 4, 1, "", "negative_adversarial_scale"]], "besskge.loss.MarginRankingLoss": [[29, 2, 1, "", "forward"], [29, 2, 1, "", "get_negative_weights"], [29, 4, 1, "", "loss_scale"], [29, 4, 1, "", "negative_adversarial_sampling"], [29, 4, 1, "", "negative_adversarial_scale"]], "besskge.loss.SampledSoftmaxCrossEntropyLoss": [[30, 2, 1, "", "forward"], [30, 2, 1, "", "get_negative_weights"], [30, 4, 1, "", "loss_scale"], [30, 4, 1, "", "negative_adversarial_sampling"], [30, 4, 1, "", "negative_adversarial_scale"]], "besskge.metric": [[32, 1, 1, "", "BaseMetric"], [33, 1, 1, "", "Evaluation"], [34, 1, 1, "", "HitsAtK"], [35, 6, 1, "", "METRICS_DICT"], [36, 1, 1, "", "ReciprocalRank"]], "besskge.metric.Evaluation": [[33, 2, 1, "", "dict_metrics_from_ranks"], [33, 2, 1, "", "ranks_from_indices"], [33, 2, 1, "", "ranks_from_scores"], [33, 2, 1, "", "stacked_metrics_from_ranks"]], "besskge.negative_sampler": [[38, 1, 1, "", "PlaceholderNegativeSampler"], [39, 1, 1, "", "RandomShardedNegativeSampler"], [40, 1, 1, "", "ShardedNegativeSampler"], [41, 1, 1, "", "TripleBasedShardedNegativeSampler"], [42, 1, 1, "", "TypeBasedShardedNegativeSampler"]], "besskge.negative_sampler.PlaceholderNegativeSampler": [[38, 4, 1, "", "corruption_scheme"], [38, 4, 1, "", "flat_negative_format"], [38, 4, 1, "", "local_sampling"], [38, 4, 1, "", "rng"]], "besskge.negative_sampler.RandomShardedNegativeSampler": [[39, 4, 1, "", "corruption_scheme"], [39, 4, 1, "", "flat_negative_format"], [39, 4, 1, "", "local_sampling"], [39, 4, 1, "", "rng"]], "besskge.negative_sampler.ShardedNegativeSampler": [[40, 4, 1, "", "corruption_scheme"], [40, 4, 1, "", "flat_negative_format"], [40, 4, 1, "", "local_sampling"], [40, 4, 1, "", "rng"]], "besskge.negative_sampler.TripleBasedShardedNegativeSampler": [[41, 4, 1, "", "corruption_scheme"], [41, 4, 1, "", "flat_negative_format"], [41, 4, 1, "", "local_sampling"], [41, 2, 1, "", "pad_negatives"], [41, 4, 1, "", "rng"], [41, 2, 1, "", "shard_negatives"]], "besskge.negative_sampler.TypeBasedShardedNegativeSampler": [[42, 4, 1, "", "corruption_scheme"], [42, 4, 1, "", "flat_negative_format"], [42, 4, 1, "", "local_sampling"], [42, 4, 1, "", "rng"]], "besskge.pipeline": [[44, 1, 1, "", "AllScoresPipeline"]], "besskge.pipeline.AllScoresPipeline": [[44, 2, 1, "", "forward"]], "besskge.scoring": [[46, 1, 1, "", "BaseScoreFunction"], [47, 1, 1, "", "BoxE"], [48, 1, 1, "", "ComplEx"], [49, 1, 1, "", "ConvE"], [50, 1, 1, "", "DistMult"], [51, 1, 1, "", "DistanceBasedScoreFunction"], [52, 1, 1, "", "InterHT"], [53, 1, 1, "", "MatrixDecompositionScoreFunction"], [54, 1, 1, "", "PairRE"], [55, 1, 1, "", "RotatE"], [56, 1, 1, "", "TranS"], [57, 1, 1, "", "TransE"], [58, 1, 1, "", "TripleRE"]], "besskge.scoring.BaseScoreFunction": [[46, 4, 1, "", "entity_embedding"], [46, 2, 1, "", "forward"], [46, 4, 1, "", "negative_sample_sharing"], [46, 4, 1, "", "relation_embedding"], [46, 2, 1, "", "score_heads"], [46, 2, 1, "", "score_tails"], [46, 2, 1, "", "score_triple"], [46, 4, 1, "", "sharding"], [46, 2, 1, "", "update_sharding"]], "besskge.scoring.BoxE": [[47, 2, 1, "", "boxe_score"], [47, 2, 1, "", "broadcasted_distance"], [47, 4, 1, "", "entity_embedding"], [47, 2, 1, "", "forward"], [47, 4, 1, "", "negative_sample_sharing"], [47, 2, 1, "", "reduce_embedding"], [47, 4, 1, "", "relation_embedding"], [47, 2, 1, "", "score_heads"], [47, 2, 1, "", "score_tails"], [47, 2, 1, "", "score_triple"], [47, 4, 1, "", "sharding"], [47, 2, 1, "", "update_sharding"]], "besskge.scoring.ComplEx": [[48, 2, 1, "", "broadcasted_dot_product"], [48, 4, 1, "", "entity_embedding"], [48, 2, 1, "", "forward"], [48, 4, 1, "", "negative_sample_sharing"], [48, 2, 1, "", "reduce_embedding"], [48, 4, 1, "", "relation_embedding"], [48, 2, 1, "", "score_heads"], [48, 2, 1, "", "score_tails"], [48, 2, 1, "", "score_triple"], [48, 4, 1, "", "sharding"], [48, 2, 1, "", "update_sharding"]], "besskge.scoring.ConvE": [[49, 2, 1, "", "broadcasted_dot_product"], [49, 4, 1, "", "entity_embedding"], [49, 2, 1, "", "forward"], [49, 4, 1, "", "negative_sample_sharing"], [49, 2, 1, "", "reduce_embedding"], [49, 4, 1, "", "relation_embedding"], [49, 2, 1, "", "score_heads"], [49, 2, 1, "", "score_tails"], [49, 2, 1, "", "score_triple"], [49, 4, 1, "", "sharding"], [49, 2, 1, "", "update_sharding"]], "besskge.scoring.DistMult": [[50, 2, 1, "", "broadcasted_dot_product"], [50, 4, 1, "", "entity_embedding"], [50, 2, 1, "", "forward"], [50, 4, 1, "", "negative_sample_sharing"], [50, 2, 1, "", "reduce_embedding"], [50, 4, 1, "", "relation_embedding"], [50, 2, 1, "", "score_heads"], [50, 2, 1, "", "score_tails"], [50, 2, 1, "", "score_triple"], [50, 4, 1, "", "sharding"], [50, 2, 1, "", "update_sharding"]], "besskge.scoring.DistanceBasedScoreFunction": [[51, 2, 1, "", "broadcasted_distance"], [51, 4, 1, "", "entity_embedding"], [51, 2, 1, "", "forward"], [51, 4, 1, "", "negative_sample_sharing"], [51, 2, 1, "", "reduce_embedding"], [51, 4, 1, "", "relation_embedding"], [51, 2, 1, "", "score_heads"], [51, 2, 1, "", "score_tails"], [51, 2, 1, "", "score_triple"], [51, 4, 1, "", "sharding"], [51, 2, 1, "", "update_sharding"]], "besskge.scoring.InterHT": [[52, 2, 1, "", "broadcasted_distance"], [52, 4, 1, "", "entity_embedding"], [52, 2, 1, "", "forward"], [52, 4, 1, "", "negative_sample_sharing"], [52, 2, 1, "", "reduce_embedding"], [52, 4, 1, "", "relation_embedding"], [52, 2, 1, "", "score_heads"], [52, 2, 1, "", "score_tails"], [52, 2, 1, "", "score_triple"], [52, 4, 1, "", "sharding"], [52, 2, 1, "", "update_sharding"]], "besskge.scoring.MatrixDecompositionScoreFunction": [[53, 2, 1, "", "broadcasted_dot_product"], [53, 4, 1, "", "entity_embedding"], [53, 2, 1, "", "forward"], [53, 4, 1, "", "negative_sample_sharing"], [53, 2, 1, "", "reduce_embedding"], [53, 4, 1, "", "relation_embedding"], [53, 2, 1, "", "score_heads"], [53, 2, 1, "", "score_tails"], [53, 2, 1, "", "score_triple"], [53, 4, 1, "", "sharding"], [53, 2, 1, "", "update_sharding"]], "besskge.scoring.PairRE": [[54, 2, 1, "", "broadcasted_distance"], [54, 4, 1, "", "entity_embedding"], [54, 2, 1, "", "forward"], [54, 4, 1, "", "negative_sample_sharing"], [54, 2, 1, "", "reduce_embedding"], [54, 4, 1, "", "relation_embedding"], [54, 2, 1, "", "score_heads"], [54, 2, 1, "", "score_tails"], [54, 2, 1, "", "score_triple"], [54, 4, 1, "", "sharding"], [54, 2, 1, "", "update_sharding"]], "besskge.scoring.RotatE": [[55, 2, 1, "", "broadcasted_distance"], [55, 4, 1, "", "entity_embedding"], [55, 2, 1, "", "forward"], [55, 4, 1, "", "negative_sample_sharing"], [55, 2, 1, "", "reduce_embedding"], [55, 4, 1, "", "relation_embedding"], [55, 2, 1, "", "score_heads"], [55, 2, 1, "", "score_tails"], [55, 2, 1, "", "score_triple"], [55, 4, 1, "", "sharding"], [55, 2, 1, "", "update_sharding"]], "besskge.scoring.TranS": [[56, 2, 1, "", "broadcasted_distance"], [56, 4, 1, "", "entity_embedding"], [56, 2, 1, "", "forward"], [56, 4, 1, "", "negative_sample_sharing"], [56, 2, 1, "", "reduce_embedding"], [56, 4, 1, "", "relation_embedding"], [56, 2, 1, "", "score_heads"], [56, 2, 1, "", "score_tails"], [56, 2, 1, "", "score_triple"], [56, 4, 1, "", "sharding"], [56, 2, 1, "", "update_sharding"]], "besskge.scoring.TransE": [[57, 2, 1, "", "broadcasted_distance"], [57, 4, 1, "", "entity_embedding"], [57, 2, 1, "", "forward"], [57, 4, 1, "", "negative_sample_sharing"], [57, 2, 1, "", "reduce_embedding"], [57, 4, 1, "", "relation_embedding"], [57, 2, 1, "", "score_heads"], [57, 2, 1, "", "score_tails"], [57, 2, 1, "", "score_triple"], [57, 4, 1, "", "sharding"], [57, 2, 1, "", "update_sharding"]], "besskge.scoring.TripleRE": [[58, 2, 1, "", "broadcasted_distance"], [58, 4, 1, "", "entity_embedding"], [58, 2, 1, "", "forward"], [58, 4, 1, "", "negative_sample_sharing"], [58, 2, 1, "", "reduce_embedding"], [58, 4, 1, "", "relation_embedding"], [58, 2, 1, "", "score_heads"], [58, 2, 1, "", "score_tails"], [58, 2, 1, "", "score_triple"], [58, 4, 1, "", "sharding"], [58, 2, 1, "", "update_sharding"]], "besskge.sharding": [[60, 1, 1, "", "PartitionedTripleSet"], [61, 1, 1, "", "Sharding"]], "besskge.sharding.PartitionedTripleSet": [[60, 2, 1, "", "create_from_dataset"], [60, 2, 1, "", "create_from_queries"], [60, 4, 1, "", "dummy"], [60, 4, 1, "", "inverse_triples"], [60, 4, 1, "", "neg_heads"], [60, 4, 1, "", "neg_tails"], [60, 4, 1, "", "partition_mode"], [60, 4, 1, "", "sharding"], [60, 4, 1, "", "triple_counts"], [60, 4, 1, "", "triple_offsets"], [60, 4, 1, "", "triple_sort_idx"], [60, 4, 1, "", "triples"], [60, 4, 1, "", "types"]], "besskge.sharding.Sharding": [[61, 2, 1, "", "create"], [61, 4, 1, "", "entity_to_idx"], [61, 4, 1, "", "entity_to_shard"], [61, 4, 1, "", "entity_type_counts"], [61, 4, 1, "", "entity_type_offsets"], [61, 2, 1, "", "load"], [61, 3, 1, "", "max_entity_per_shard"], [61, 3, 1, "", "n_entity"], [61, 4, 1, "", "n_shard"], [61, 2, 1, "", "save"], [61, 4, 1, "", "shard_and_idx_to_entity"], [61, 4, 1, "", "shard_counts"]], "besskge.utils": [[63, 5, 1, "", "complex_multiplication"], [64, 5, 1, "", "complex_rotation"], [65, 5, 1, "", "gather_indices"], [66, 5, 1, "", "get_entity_filter"]]}, "objtypes": {"0": "py:module", "1": "py:class", "2": "py:method", "3": "py:property", "4": "py:attribute", "5": "py:function", "6": "py:data"}, "objnames": {"0": ["py", "module", "Python module"], "1": ["py", "class", "Python class"], "2": ["py", "method", "Python method"], "3": ["py", "property", "Python property"], "4": ["py", "attribute", "Python attribute"], "5": ["py", "function", "Python function"], "6": ["py", "data", "Python data"]}, "titleterms": {"bess": [0, 1, 3, 4, 9, 10, 11, 12, 13, 14, 67], "kge": [0, 3, 4, 67], "api": 0, "refer": 0, "overview": 1, "bibliographi": 2, "how": [3, 4], "contribut": [3, 4], "project": [3, 4], "v": [3, 4], "code": [3, 4], "server": [3, 4], "paperspac": [3, 4], "setup": [3, 4], "local": [3, 4], "machin": [3, 4], "develop": [3, 4], "tip": [3, 4], "besskg": [5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66], "batch_sampl": [5, 6, 7, 8], "randomshardedbatchsampl": 6, "rigidshardedbatchsampl": 7, "shardedbatchsampl": 8, "allscoresbess": 10, "embeddingmovingbesskg": 12, "scoremovingbesskg": 13, "topkquerybesskg": 14, "dataset": [15, 16], "kgdataset": 16, "embed": [17, 18, 19, 20, 21, 22, 23, 24], "init_kge_norm": 18, "init_kge_uniform": 19, "init_uniform_norm": 20, "init_xavier_norm": 21, "initialize_entity_embed": 22, "initialize_relation_embed": 23, "refactor_embedding_shard": 24, "loss": [25, 26, 27, 28, 29, 30], "baselossfunct": 26, "logsigmoidloss": 27, "marginbasedlossfunct": 28, "marginrankingloss": 29, "sampledsoftmaxcrossentropyloss": 30, "metric": [31, 32, 33, 34, 35, 36], "basemetr": 32, "evalu": 33, "hitsatk": 34, "metrics_dict": 35, "reciprocalrank": 36, "negative_sampl": [37, 38, 39, 40, 41, 42], "placeholdernegativesampl": 38, "randomshardednegativesampl": 39, "shardednegativesampl": 40, "triplebasedshardednegativesampl": 41, "typebasedshardednegativesampl": 42, "pipelin": [43, 44], "allscorespipelin": 44, "score": [45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58], "basescorefunct": 46, "box": 47, "complex": 48, "conv": 49, "distmult": 50, "distancebasedscorefunct": 51, "interht": 52, "matrixdecompositionscorefunct": 53, "pairr": 54, "rotat": 55, "tran": 56, "trans": 57, "tripler": 58, "shard": [59, 60, 61], "partitionedtripleset": 60, "util": [62, 63, 64, 65, 66], "complex_multipl": 63, "complex_rot": 64, "gather_indic": 65, "get_entity_filt": 66, "content": 67, "user": 68, "guid": 68, "instal": 68, "usag": 68, "get": 68, "start": 68, "limit": 68}, "envversion": {"sphinx.domains.c": 2, "sphinx.domains.changeset": 1, "sphinx.domains.citation": 1, "sphinx.domains.cpp": 8, "sphinx.domains.index": 1, "sphinx.domains.javascript": 2, "sphinx.domains.math": 2, "sphinx.domains.python": 3, "sphinx.domains.rst": 2, "sphinx.domains.std": 2, "sphinx.ext.todo": 2, "sphinx.ext.viewcode": 1, "sphinx.ext.intersphinx": 1, "sphinxcontrib.bibtex": 9, "sphinx": 57}, "alltitles": {"BESS-KGE API Reference": [[0, "bess-kge-api-reference"]], "BESS overview": [[1, "bess-overview"]], "Bibliography": [[2, "bibliography"]], "How to contribute to the BESS-KGE project": [[3, "how-to-contribute-to-the-bess-kge-project"], [4, "how-to-contribute-to-the-bess-kge-project"]], "VS Code server on Paperspace": [[3, "vs-code-server-on-paperspace"], [4, "vs-code-server-on-paperspace"]], "Setup on local machine": [[3, "setup-on-local-machine"], [4, "setup-on-local-machine"]], "Development tips": [[3, "development-tips"], [4, "development-tips"]], "besskge.batch_sampler": [[5, "module-besskge.batch_sampler"]], "besskge.batch_sampler.RandomShardedBatchSampler": [[6, "besskge-batch-sampler-randomshardedbatchsampler"]], "besskge.batch_sampler.RigidShardedBatchSampler": [[7, "besskge-batch-sampler-rigidshardedbatchsampler"]], "besskge.batch_sampler.ShardedBatchSampler": [[8, "besskge-batch-sampler-shardedbatchsampler"]], "besskge.bess": [[9, "module-besskge.bess"]], "besskge.bess.AllScoresBESS": [[10, "besskge-bess-allscoresbess"]], "besskge.bess.BessKGE": [[11, "besskge-bess-besskge"]], "besskge.bess.EmbeddingMovingBessKGE": [[12, "besskge-bess-embeddingmovingbesskge"]], "besskge.bess.ScoreMovingBessKGE": [[13, "besskge-bess-scoremovingbesskge"]], "besskge.bess.TopKQueryBessKGE": [[14, "besskge-bess-topkquerybesskge"]], "besskge.dataset": [[15, "module-besskge.dataset"]], "besskge.dataset.KGDataset": [[16, "besskge-dataset-kgdataset"]], "besskge.embedding": [[17, "module-besskge.embedding"]], "besskge.embedding.init_KGE_normal": [[18, "besskge-embedding-init-kge-normal"]], "besskge.embedding.init_KGE_uniform": [[19, "besskge-embedding-init-kge-uniform"]], "besskge.embedding.init_uniform_norm": [[20, "besskge-embedding-init-uniform-norm"]], "besskge.embedding.init_xavier_norm": [[21, "besskge-embedding-init-xavier-norm"]], "besskge.embedding.initialize_entity_embedding": [[22, "besskge-embedding-initialize-entity-embedding"]], "besskge.embedding.initialize_relation_embedding": [[23, "besskge-embedding-initialize-relation-embedding"]], "besskge.embedding.refactor_embedding_sharding": [[24, "besskge-embedding-refactor-embedding-sharding"]], "besskge.loss": [[25, "module-besskge.loss"]], "besskge.loss.BaseLossFunction": [[26, "besskge-loss-baselossfunction"]], "besskge.loss.LogSigmoidLoss": [[27, "besskge-loss-logsigmoidloss"]], "besskge.loss.MarginBasedLossFunction": [[28, "besskge-loss-marginbasedlossfunction"]], "besskge.loss.MarginRankingLoss": [[29, "besskge-loss-marginrankingloss"]], "besskge.loss.SampledSoftmaxCrossEntropyLoss": [[30, "besskge-loss-sampledsoftmaxcrossentropyloss"]], "besskge.metric": [[31, "module-besskge.metric"]], "besskge.metric.BaseMetric": [[32, "besskge-metric-basemetric"]], "besskge.metric.Evaluation": [[33, "besskge-metric-evaluation"]], "besskge.metric.HitsAtK": [[34, "besskge-metric-hitsatk"]], "besskge.metric.METRICS_DICT": [[35, "besskge-metric-metrics-dict"]], "besskge.metric.ReciprocalRank": [[36, "besskge-metric-reciprocalrank"]], "besskge.negative_sampler": [[37, "module-besskge.negative_sampler"]], "besskge.negative_sampler.PlaceholderNegativeSampler": [[38, "besskge-negative-sampler-placeholdernegativesampler"]], "besskge.negative_sampler.RandomShardedNegativeSampler": [[39, "besskge-negative-sampler-randomshardednegativesampler"]], "besskge.negative_sampler.ShardedNegativeSampler": [[40, "besskge-negative-sampler-shardednegativesampler"]], "besskge.negative_sampler.TripleBasedShardedNegativeSampler": [[41, "besskge-negative-sampler-triplebasedshardednegativesampler"]], "besskge.negative_sampler.TypeBasedShardedNegativeSampler": [[42, "besskge-negative-sampler-typebasedshardednegativesampler"]], "besskge.pipeline": [[43, "module-besskge.pipeline"]], "besskge.pipeline.AllScoresPipeline": [[44, "besskge-pipeline-allscorespipeline"]], "besskge.scoring": [[45, "module-besskge.scoring"]], "besskge.scoring.BaseScoreFunction": [[46, "besskge-scoring-basescorefunction"]], "besskge.scoring.BoxE": [[47, "besskge-scoring-boxe"]], "besskge.scoring.ComplEx": [[48, "besskge-scoring-complex"]], "besskge.scoring.ConvE": [[49, "besskge-scoring-conve"]], "besskge.scoring.DistMult": [[50, "besskge-scoring-distmult"]], "besskge.scoring.DistanceBasedScoreFunction": [[51, "besskge-scoring-distancebasedscorefunction"]], "besskge.scoring.InterHT": [[52, "besskge-scoring-interht"]], "besskge.scoring.MatrixDecompositionScoreFunction": [[53, "besskge-scoring-matrixdecompositionscorefunction"]], "besskge.scoring.PairRE": [[54, "besskge-scoring-pairre"]], "besskge.scoring.RotatE": [[55, "besskge-scoring-rotate"]], "besskge.scoring.TranS": [[56, "besskge-scoring-trans"]], "besskge.scoring.TransE": [[57, "besskge-scoring-transe"]], "besskge.scoring.TripleRE": [[58, "besskge-scoring-triplere"]], "besskge.sharding": [[59, "module-besskge.sharding"]], "besskge.sharding.PartitionedTripleSet": [[60, "besskge-sharding-partitionedtripleset"]], "besskge.sharding.Sharding": [[61, "besskge-sharding-sharding"]], "besskge.utils": [[62, "module-besskge.utils"]], "besskge.utils.complex_multiplication": [[63, "besskge-utils-complex-multiplication"]], "besskge.utils.complex_rotation": [[64, "besskge-utils-complex-rotation"]], "besskge.utils.gather_indices": [[65, "besskge-utils-gather-indices"]], "besskge.utils.get_entity_filter": [[66, "besskge-utils-get-entity-filter"]], "BESS-KGE": [[67, "module-besskge"]], "Contents": [[67, null]], "User guide": [[68, "user-guide"]], "Installation and usage": [[68, "installation-and-usage"]], "Getting started": [[68, "getting-started"]], "Limitations": [[68, "limitations"]]}, "indexentries": {"besskge.batch_sampler": [[5, "module-besskge.batch_sampler"]], "module": [[5, "module-besskge.batch_sampler"], [9, "module-besskge.bess"], [15, "module-besskge.dataset"], [17, "module-besskge.embedding"], [25, "module-besskge.loss"], [31, "module-besskge.metric"], [37, "module-besskge.negative_sampler"], [43, "module-besskge.pipeline"], [45, "module-besskge.scoring"], [59, "module-besskge.sharding"], [62, "module-besskge.utils"], [67, "module-besskge"]], "randomshardedbatchsampler (class in besskge.batch_sampler)": [[6, "besskge.batch_sampler.RandomShardedBatchSampler"]], "get_dataloader() (besskge.batch_sampler.randomshardedbatchsampler method)": [[6, "besskge.batch_sampler.RandomShardedBatchSampler.get_dataloader"]], "get_dataloader_sampler() (besskge.batch_sampler.randomshardedbatchsampler method)": [[6, "besskge.batch_sampler.RandomShardedBatchSampler.get_dataloader_sampler"]], "sample_triples() (besskge.batch_sampler.randomshardedbatchsampler method)": [[6, "besskge.batch_sampler.RandomShardedBatchSampler.sample_triples"]], "worker_init_fn() (besskge.batch_sampler.randomshardedbatchsampler static method)": [[6, "besskge.batch_sampler.RandomShardedBatchSampler.worker_init_fn"]], "rigidshardedbatchsampler (class in besskge.batch_sampler)": [[7, "besskge.batch_sampler.RigidShardedBatchSampler"]], "get_dataloader() (besskge.batch_sampler.rigidshardedbatchsampler method)": [[7, "besskge.batch_sampler.RigidShardedBatchSampler.get_dataloader"]], "get_dataloader_sampler() (besskge.batch_sampler.rigidshardedbatchsampler method)": [[7, "besskge.batch_sampler.RigidShardedBatchSampler.get_dataloader_sampler"]], "sample_triples() (besskge.batch_sampler.rigidshardedbatchsampler method)": [[7, "besskge.batch_sampler.RigidShardedBatchSampler.sample_triples"]], "worker_init_fn() (besskge.batch_sampler.rigidshardedbatchsampler static method)": [[7, "besskge.batch_sampler.RigidShardedBatchSampler.worker_init_fn"]], "shardedbatchsampler (class in besskge.batch_sampler)": [[8, "besskge.batch_sampler.ShardedBatchSampler"]], "get_dataloader() (besskge.batch_sampler.shardedbatchsampler method)": [[8, "besskge.batch_sampler.ShardedBatchSampler.get_dataloader"]], "get_dataloader_sampler() (besskge.batch_sampler.shardedbatchsampler method)": [[8, "besskge.batch_sampler.ShardedBatchSampler.get_dataloader_sampler"]], "sample_triples() (besskge.batch_sampler.shardedbatchsampler method)": [[8, "besskge.batch_sampler.ShardedBatchSampler.sample_triples"]], "worker_init_fn() (besskge.batch_sampler.shardedbatchsampler static method)": [[8, "besskge.batch_sampler.ShardedBatchSampler.worker_init_fn"]], "besskge.bess": [[9, "module-besskge.bess"]], "allscoresbess (class in besskge.bess)": [[10, "besskge.bess.AllScoresBESS"]], "forward() (besskge.bess.allscoresbess method)": [[10, "besskge.bess.AllScoresBESS.forward"]], "besskge (class in besskge.bess)": [[11, "besskge.bess.BessKGE"]], "forward() (besskge.bess.besskge method)": [[11, "besskge.bess.BessKGE.forward"]], "n_embedding_parameters (besskge.bess.besskge property)": [[11, "besskge.bess.BessKGE.n_embedding_parameters"]], "score_batch() (besskge.bess.besskge method)": [[11, "besskge.bess.BessKGE.score_batch"]], "embeddingmovingbesskge (class in besskge.bess)": [[12, "besskge.bess.EmbeddingMovingBessKGE"]], "forward() (besskge.bess.embeddingmovingbesskge method)": [[12, "besskge.bess.EmbeddingMovingBessKGE.forward"]], "n_embedding_parameters (besskge.bess.embeddingmovingbesskge property)": [[12, "besskge.bess.EmbeddingMovingBessKGE.n_embedding_parameters"]], "score_batch() (besskge.bess.embeddingmovingbesskge method)": [[12, "besskge.bess.EmbeddingMovingBessKGE.score_batch"]], "scoremovingbesskge (class in besskge.bess)": [[13, "besskge.bess.ScoreMovingBessKGE"]], "forward() (besskge.bess.scoremovingbesskge method)": [[13, "besskge.bess.ScoreMovingBessKGE.forward"]], "n_embedding_parameters (besskge.bess.scoremovingbesskge property)": [[13, "besskge.bess.ScoreMovingBessKGE.n_embedding_parameters"]], "score_batch() (besskge.bess.scoremovingbesskge method)": [[13, "besskge.bess.ScoreMovingBessKGE.score_batch"]], "topkquerybesskge (class in besskge.bess)": [[14, "besskge.bess.TopKQueryBessKGE"]], "forward() (besskge.bess.topkquerybesskge method)": [[14, "besskge.bess.TopKQueryBessKGE.forward"]], "besskge.dataset": [[15, "module-besskge.dataset"]], "kgdataset (class in besskge.dataset)": [[16, "besskge.dataset.KGDataset"]], "build_ogbl_biokg() (besskge.dataset.kgdataset class method)": [[16, "besskge.dataset.KGDataset.build_ogbl_biokg"]], "build_ogbl_wikikg2() (besskge.dataset.kgdataset class method)": [[16, "besskge.dataset.KGDataset.build_ogbl_wikikg2"]], "build_openbiolink() (besskge.dataset.kgdataset class method)": [[16, "besskge.dataset.KGDataset.build_openbiolink"]], "build_yago310() (besskge.dataset.kgdataset class method)": [[16, "besskge.dataset.KGDataset.build_yago310"]], "entity_dict (besskge.dataset.kgdataset attribute)": [[16, "besskge.dataset.KGDataset.entity_dict"]], "from_dataframe() (besskge.dataset.kgdataset class method)": [[16, "besskge.dataset.KGDataset.from_dataframe"]], "from_triples() (besskge.dataset.kgdataset class method)": [[16, "besskge.dataset.KGDataset.from_triples"]], "ht_types (besskge.dataset.kgdataset property)": [[16, "besskge.dataset.KGDataset.ht_types"]], "load() (besskge.dataset.kgdataset class method)": [[16, "besskge.dataset.KGDataset.load"]], "n_entity (besskge.dataset.kgdataset attribute)": [[16, "besskge.dataset.KGDataset.n_entity"]], "n_relation_type (besskge.dataset.kgdataset attribute)": [[16, "besskge.dataset.KGDataset.n_relation_type"]], "neg_heads (besskge.dataset.kgdataset attribute)": [[16, "besskge.dataset.KGDataset.neg_heads"]], "neg_tails (besskge.dataset.kgdataset attribute)": [[16, "besskge.dataset.KGDataset.neg_tails"]], "original_triple_ids (besskge.dataset.kgdataset attribute)": [[16, "besskge.dataset.KGDataset.original_triple_ids"]], "relation_dict (besskge.dataset.kgdataset attribute)": [[16, "besskge.dataset.KGDataset.relation_dict"]], "save() (besskge.dataset.kgdataset method)": [[16, "besskge.dataset.KGDataset.save"]], "triples (besskge.dataset.kgdataset attribute)": [[16, "besskge.dataset.KGDataset.triples"]], "type_offsets (besskge.dataset.kgdataset attribute)": [[16, "besskge.dataset.KGDataset.type_offsets"]], "besskge.embedding": [[17, "module-besskge.embedding"]], "init_kge_normal() (in module besskge.embedding)": [[18, "besskge.embedding.init_KGE_normal"]], "init_kge_uniform() (in module besskge.embedding)": [[19, "besskge.embedding.init_KGE_uniform"]], "init_uniform_norm() (in module besskge.embedding)": [[20, "besskge.embedding.init_uniform_norm"]], "init_xavier_norm() (in module besskge.embedding)": [[21, "besskge.embedding.init_xavier_norm"]], "initialize_entity_embedding() (in module besskge.embedding)": [[22, "besskge.embedding.initialize_entity_embedding"]], "initialize_relation_embedding() (in module besskge.embedding)": [[23, "besskge.embedding.initialize_relation_embedding"]], "refactor_embedding_sharding() (in module besskge.embedding)": [[24, "besskge.embedding.refactor_embedding_sharding"]], "besskge.loss": [[25, "module-besskge.loss"]], "baselossfunction (class in besskge.loss)": [[26, "besskge.loss.BaseLossFunction"]], "forward() (besskge.loss.baselossfunction method)": [[26, "besskge.loss.BaseLossFunction.forward"]], "get_negative_weights() (besskge.loss.baselossfunction method)": [[26, "besskge.loss.BaseLossFunction.get_negative_weights"]], "loss_scale (besskge.loss.baselossfunction attribute)": [[26, "besskge.loss.BaseLossFunction.loss_scale"]], "negative_adversarial_sampling (besskge.loss.baselossfunction attribute)": [[26, "besskge.loss.BaseLossFunction.negative_adversarial_sampling"]], "negative_adversarial_scale (besskge.loss.baselossfunction attribute)": [[26, "besskge.loss.BaseLossFunction.negative_adversarial_scale"]], "logsigmoidloss (class in besskge.loss)": [[27, "besskge.loss.LogSigmoidLoss"]], "forward() (besskge.loss.logsigmoidloss method)": [[27, "besskge.loss.LogSigmoidLoss.forward"]], "get_negative_weights() (besskge.loss.logsigmoidloss method)": [[27, "besskge.loss.LogSigmoidLoss.get_negative_weights"]], "loss_scale (besskge.loss.logsigmoidloss attribute)": [[27, "besskge.loss.LogSigmoidLoss.loss_scale"]], "negative_adversarial_sampling (besskge.loss.logsigmoidloss attribute)": [[27, "besskge.loss.LogSigmoidLoss.negative_adversarial_sampling"]], "negative_adversarial_scale (besskge.loss.logsigmoidloss attribute)": [[27, "besskge.loss.LogSigmoidLoss.negative_adversarial_scale"]], "marginbasedlossfunction (class in besskge.loss)": [[28, "besskge.loss.MarginBasedLossFunction"]], "forward() (besskge.loss.marginbasedlossfunction method)": [[28, "besskge.loss.MarginBasedLossFunction.forward"]], "get_negative_weights() (besskge.loss.marginbasedlossfunction method)": [[28, "besskge.loss.MarginBasedLossFunction.get_negative_weights"]], "loss_scale (besskge.loss.marginbasedlossfunction attribute)": [[28, "besskge.loss.MarginBasedLossFunction.loss_scale"]], "negative_adversarial_sampling (besskge.loss.marginbasedlossfunction attribute)": [[28, "besskge.loss.MarginBasedLossFunction.negative_adversarial_sampling"]], "negative_adversarial_scale (besskge.loss.marginbasedlossfunction attribute)": [[28, "besskge.loss.MarginBasedLossFunction.negative_adversarial_scale"]], "marginrankingloss (class in besskge.loss)": [[29, "besskge.loss.MarginRankingLoss"]], "forward() (besskge.loss.marginrankingloss method)": [[29, "besskge.loss.MarginRankingLoss.forward"]], "get_negative_weights() (besskge.loss.marginrankingloss method)": [[29, "besskge.loss.MarginRankingLoss.get_negative_weights"]], "loss_scale (besskge.loss.marginrankingloss attribute)": [[29, "besskge.loss.MarginRankingLoss.loss_scale"]], "negative_adversarial_sampling (besskge.loss.marginrankingloss attribute)": [[29, "besskge.loss.MarginRankingLoss.negative_adversarial_sampling"]], "negative_adversarial_scale (besskge.loss.marginrankingloss attribute)": [[29, "besskge.loss.MarginRankingLoss.negative_adversarial_scale"]], "sampledsoftmaxcrossentropyloss (class in besskge.loss)": [[30, "besskge.loss.SampledSoftmaxCrossEntropyLoss"]], "forward() (besskge.loss.sampledsoftmaxcrossentropyloss method)": [[30, "besskge.loss.SampledSoftmaxCrossEntropyLoss.forward"]], "get_negative_weights() (besskge.loss.sampledsoftmaxcrossentropyloss method)": [[30, "besskge.loss.SampledSoftmaxCrossEntropyLoss.get_negative_weights"]], "loss_scale (besskge.loss.sampledsoftmaxcrossentropyloss attribute)": [[30, "besskge.loss.SampledSoftmaxCrossEntropyLoss.loss_scale"]], "negative_adversarial_sampling (besskge.loss.sampledsoftmaxcrossentropyloss attribute)": [[30, "besskge.loss.SampledSoftmaxCrossEntropyLoss.negative_adversarial_sampling"]], "negative_adversarial_scale (besskge.loss.sampledsoftmaxcrossentropyloss attribute)": [[30, "besskge.loss.SampledSoftmaxCrossEntropyLoss.negative_adversarial_scale"]], "besskge.metric": [[31, "module-besskge.metric"]], "basemetric (class in besskge.metric)": [[32, "besskge.metric.BaseMetric"]], "evaluation (class in besskge.metric)": [[33, "besskge.metric.Evaluation"]], "dict_metrics_from_ranks() (besskge.metric.evaluation method)": [[33, "besskge.metric.Evaluation.dict_metrics_from_ranks"]], "ranks_from_indices() (besskge.metric.evaluation method)": [[33, "besskge.metric.Evaluation.ranks_from_indices"]], "ranks_from_scores() (besskge.metric.evaluation method)": [[33, "besskge.metric.Evaluation.ranks_from_scores"]], "stacked_metrics_from_ranks() (besskge.metric.evaluation method)": [[33, "besskge.metric.Evaluation.stacked_metrics_from_ranks"]], "hitsatk (class in besskge.metric)": [[34, "besskge.metric.HitsAtK"]], "metrics_dict (in module besskge.metric)": [[35, "besskge.metric.METRICS_DICT"]], "reciprocalrank (class in besskge.metric)": [[36, "besskge.metric.ReciprocalRank"]], "besskge.negative_sampler": [[37, "module-besskge.negative_sampler"]], "placeholdernegativesampler (class in besskge.negative_sampler)": [[38, "besskge.negative_sampler.PlaceholderNegativeSampler"]], "corruption_scheme (besskge.negative_sampler.placeholdernegativesampler attribute)": [[38, "besskge.negative_sampler.PlaceholderNegativeSampler.corruption_scheme"]], "flat_negative_format (besskge.negative_sampler.placeholdernegativesampler attribute)": [[38, "besskge.negative_sampler.PlaceholderNegativeSampler.flat_negative_format"]], "local_sampling (besskge.negative_sampler.placeholdernegativesampler attribute)": [[38, "besskge.negative_sampler.PlaceholderNegativeSampler.local_sampling"]], "rng (besskge.negative_sampler.placeholdernegativesampler attribute)": [[38, "besskge.negative_sampler.PlaceholderNegativeSampler.rng"]], "randomshardednegativesampler (class in besskge.negative_sampler)": [[39, "besskge.negative_sampler.RandomShardedNegativeSampler"]], "corruption_scheme (besskge.negative_sampler.randomshardednegativesampler attribute)": [[39, "besskge.negative_sampler.RandomShardedNegativeSampler.corruption_scheme"]], "flat_negative_format (besskge.negative_sampler.randomshardednegativesampler attribute)": [[39, "besskge.negative_sampler.RandomShardedNegativeSampler.flat_negative_format"]], "local_sampling (besskge.negative_sampler.randomshardednegativesampler attribute)": [[39, "besskge.negative_sampler.RandomShardedNegativeSampler.local_sampling"]], "rng (besskge.negative_sampler.randomshardednegativesampler attribute)": [[39, "besskge.negative_sampler.RandomShardedNegativeSampler.rng"]], "shardednegativesampler (class in besskge.negative_sampler)": [[40, "besskge.negative_sampler.ShardedNegativeSampler"]], "corruption_scheme (besskge.negative_sampler.shardednegativesampler attribute)": [[40, "besskge.negative_sampler.ShardedNegativeSampler.corruption_scheme"]], "flat_negative_format (besskge.negative_sampler.shardednegativesampler attribute)": [[40, "besskge.negative_sampler.ShardedNegativeSampler.flat_negative_format"]], "local_sampling (besskge.negative_sampler.shardednegativesampler attribute)": [[40, "besskge.negative_sampler.ShardedNegativeSampler.local_sampling"]], "rng (besskge.negative_sampler.shardednegativesampler attribute)": [[40, "besskge.negative_sampler.ShardedNegativeSampler.rng"]], "triplebasedshardednegativesampler (class in besskge.negative_sampler)": [[41, "besskge.negative_sampler.TripleBasedShardedNegativeSampler"]], "corruption_scheme (besskge.negative_sampler.triplebasedshardednegativesampler attribute)": [[41, "besskge.negative_sampler.TripleBasedShardedNegativeSampler.corruption_scheme"]], "flat_negative_format (besskge.negative_sampler.triplebasedshardednegativesampler attribute)": [[41, "besskge.negative_sampler.TripleBasedShardedNegativeSampler.flat_negative_format"]], "local_sampling (besskge.negative_sampler.triplebasedshardednegativesampler attribute)": [[41, "besskge.negative_sampler.TripleBasedShardedNegativeSampler.local_sampling"]], "pad_negatives() (besskge.negative_sampler.triplebasedshardednegativesampler method)": [[41, "besskge.negative_sampler.TripleBasedShardedNegativeSampler.pad_negatives"]], "rng (besskge.negative_sampler.triplebasedshardednegativesampler attribute)": [[41, "besskge.negative_sampler.TripleBasedShardedNegativeSampler.rng"]], "shard_negatives() (besskge.negative_sampler.triplebasedshardednegativesampler method)": [[41, "besskge.negative_sampler.TripleBasedShardedNegativeSampler.shard_negatives"]], "typebasedshardednegativesampler (class in besskge.negative_sampler)": [[42, "besskge.negative_sampler.TypeBasedShardedNegativeSampler"]], "corruption_scheme (besskge.negative_sampler.typebasedshardednegativesampler attribute)": [[42, "besskge.negative_sampler.TypeBasedShardedNegativeSampler.corruption_scheme"]], "flat_negative_format (besskge.negative_sampler.typebasedshardednegativesampler attribute)": [[42, "besskge.negative_sampler.TypeBasedShardedNegativeSampler.flat_negative_format"]], "local_sampling (besskge.negative_sampler.typebasedshardednegativesampler attribute)": [[42, "besskge.negative_sampler.TypeBasedShardedNegativeSampler.local_sampling"]], "rng (besskge.negative_sampler.typebasedshardednegativesampler attribute)": [[42, "besskge.negative_sampler.TypeBasedShardedNegativeSampler.rng"]], "besskge.pipeline": [[43, "module-besskge.pipeline"]], "allscorespipeline (class in besskge.pipeline)": [[44, "besskge.pipeline.AllScoresPipeline"]], "forward() (besskge.pipeline.allscorespipeline method)": [[44, "besskge.pipeline.AllScoresPipeline.forward"]], "besskge.scoring": [[45, "module-besskge.scoring"]], "basescorefunction (class in besskge.scoring)": [[46, "besskge.scoring.BaseScoreFunction"]], "entity_embedding (besskge.scoring.basescorefunction attribute)": [[46, "besskge.scoring.BaseScoreFunction.entity_embedding"]], "forward() (besskge.scoring.basescorefunction method)": [[46, "besskge.scoring.BaseScoreFunction.forward"]], "negative_sample_sharing (besskge.scoring.basescorefunction attribute)": [[46, "besskge.scoring.BaseScoreFunction.negative_sample_sharing"]], "relation_embedding (besskge.scoring.basescorefunction attribute)": [[46, "besskge.scoring.BaseScoreFunction.relation_embedding"]], "score_heads() (besskge.scoring.basescorefunction method)": [[46, "besskge.scoring.BaseScoreFunction.score_heads"]], "score_tails() (besskge.scoring.basescorefunction method)": [[46, "besskge.scoring.BaseScoreFunction.score_tails"]], "score_triple() (besskge.scoring.basescorefunction method)": [[46, "besskge.scoring.BaseScoreFunction.score_triple"]], "sharding (besskge.scoring.basescorefunction attribute)": [[46, "besskge.scoring.BaseScoreFunction.sharding"]], "update_sharding() (besskge.scoring.basescorefunction method)": [[46, "besskge.scoring.BaseScoreFunction.update_sharding"]], "boxe (class in besskge.scoring)": [[47, "besskge.scoring.BoxE"]], "boxe_score() (besskge.scoring.boxe method)": [[47, "besskge.scoring.BoxE.boxe_score"]], "broadcasted_distance() (besskge.scoring.boxe method)": [[47, "besskge.scoring.BoxE.broadcasted_distance"]], "entity_embedding (besskge.scoring.boxe attribute)": [[47, "besskge.scoring.BoxE.entity_embedding"]], "forward() (besskge.scoring.boxe method)": [[47, "besskge.scoring.BoxE.forward"]], "negative_sample_sharing (besskge.scoring.boxe attribute)": [[47, "besskge.scoring.BoxE.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.boxe method)": [[47, "besskge.scoring.BoxE.reduce_embedding"]], "relation_embedding (besskge.scoring.boxe attribute)": [[47, "besskge.scoring.BoxE.relation_embedding"]], "score_heads() (besskge.scoring.boxe method)": [[47, "besskge.scoring.BoxE.score_heads"]], "score_tails() (besskge.scoring.boxe method)": [[47, "besskge.scoring.BoxE.score_tails"]], "score_triple() (besskge.scoring.boxe method)": [[47, "besskge.scoring.BoxE.score_triple"]], "sharding (besskge.scoring.boxe attribute)": [[47, "besskge.scoring.BoxE.sharding"]], "update_sharding() (besskge.scoring.boxe method)": [[47, "besskge.scoring.BoxE.update_sharding"]], "complex (class in besskge.scoring)": [[48, "besskge.scoring.ComplEx"]], "broadcasted_dot_product() (besskge.scoring.complex method)": [[48, "besskge.scoring.ComplEx.broadcasted_dot_product"]], "entity_embedding (besskge.scoring.complex attribute)": [[48, "besskge.scoring.ComplEx.entity_embedding"]], "forward() (besskge.scoring.complex method)": [[48, "besskge.scoring.ComplEx.forward"]], "negative_sample_sharing (besskge.scoring.complex attribute)": [[48, "besskge.scoring.ComplEx.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.complex method)": [[48, "besskge.scoring.ComplEx.reduce_embedding"]], "relation_embedding (besskge.scoring.complex attribute)": [[48, "besskge.scoring.ComplEx.relation_embedding"]], "score_heads() (besskge.scoring.complex method)": [[48, "besskge.scoring.ComplEx.score_heads"]], "score_tails() (besskge.scoring.complex method)": [[48, "besskge.scoring.ComplEx.score_tails"]], "score_triple() (besskge.scoring.complex method)": [[48, "besskge.scoring.ComplEx.score_triple"]], "sharding (besskge.scoring.complex attribute)": [[48, "besskge.scoring.ComplEx.sharding"]], "update_sharding() (besskge.scoring.complex method)": [[48, "besskge.scoring.ComplEx.update_sharding"]], "conve (class in besskge.scoring)": [[49, "besskge.scoring.ConvE"]], "broadcasted_dot_product() (besskge.scoring.conve method)": [[49, "besskge.scoring.ConvE.broadcasted_dot_product"]], "entity_embedding (besskge.scoring.conve attribute)": [[49, "besskge.scoring.ConvE.entity_embedding"]], "forward() (besskge.scoring.conve method)": [[49, "besskge.scoring.ConvE.forward"]], "negative_sample_sharing (besskge.scoring.conve attribute)": [[49, "besskge.scoring.ConvE.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.conve method)": [[49, "besskge.scoring.ConvE.reduce_embedding"]], "relation_embedding (besskge.scoring.conve attribute)": [[49, "besskge.scoring.ConvE.relation_embedding"]], "score_heads() (besskge.scoring.conve method)": [[49, "besskge.scoring.ConvE.score_heads"]], "score_tails() (besskge.scoring.conve method)": [[49, "besskge.scoring.ConvE.score_tails"]], "score_triple() (besskge.scoring.conve method)": [[49, "besskge.scoring.ConvE.score_triple"]], "sharding (besskge.scoring.conve attribute)": [[49, "besskge.scoring.ConvE.sharding"]], "update_sharding() (besskge.scoring.conve method)": [[49, "besskge.scoring.ConvE.update_sharding"]], "distmult (class in besskge.scoring)": [[50, "besskge.scoring.DistMult"]], "broadcasted_dot_product() (besskge.scoring.distmult method)": [[50, "besskge.scoring.DistMult.broadcasted_dot_product"]], "entity_embedding (besskge.scoring.distmult attribute)": [[50, "besskge.scoring.DistMult.entity_embedding"]], "forward() (besskge.scoring.distmult method)": [[50, "besskge.scoring.DistMult.forward"]], "negative_sample_sharing (besskge.scoring.distmult attribute)": [[50, "besskge.scoring.DistMult.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.distmult method)": [[50, "besskge.scoring.DistMult.reduce_embedding"]], "relation_embedding (besskge.scoring.distmult attribute)": [[50, "besskge.scoring.DistMult.relation_embedding"]], "score_heads() (besskge.scoring.distmult method)": [[50, "besskge.scoring.DistMult.score_heads"]], "score_tails() (besskge.scoring.distmult method)": [[50, "besskge.scoring.DistMult.score_tails"]], "score_triple() (besskge.scoring.distmult method)": [[50, "besskge.scoring.DistMult.score_triple"]], "sharding (besskge.scoring.distmult attribute)": [[50, "besskge.scoring.DistMult.sharding"]], "update_sharding() (besskge.scoring.distmult method)": [[50, "besskge.scoring.DistMult.update_sharding"]], "distancebasedscorefunction (class in besskge.scoring)": [[51, "besskge.scoring.DistanceBasedScoreFunction"]], "broadcasted_distance() (besskge.scoring.distancebasedscorefunction method)": [[51, "besskge.scoring.DistanceBasedScoreFunction.broadcasted_distance"]], "entity_embedding (besskge.scoring.distancebasedscorefunction attribute)": [[51, "besskge.scoring.DistanceBasedScoreFunction.entity_embedding"]], "forward() (besskge.scoring.distancebasedscorefunction method)": [[51, "besskge.scoring.DistanceBasedScoreFunction.forward"]], "negative_sample_sharing (besskge.scoring.distancebasedscorefunction attribute)": [[51, "besskge.scoring.DistanceBasedScoreFunction.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.distancebasedscorefunction method)": [[51, "besskge.scoring.DistanceBasedScoreFunction.reduce_embedding"]], "relation_embedding (besskge.scoring.distancebasedscorefunction attribute)": [[51, "besskge.scoring.DistanceBasedScoreFunction.relation_embedding"]], "score_heads() (besskge.scoring.distancebasedscorefunction method)": [[51, "besskge.scoring.DistanceBasedScoreFunction.score_heads"]], "score_tails() (besskge.scoring.distancebasedscorefunction method)": [[51, "besskge.scoring.DistanceBasedScoreFunction.score_tails"]], "score_triple() (besskge.scoring.distancebasedscorefunction method)": [[51, "besskge.scoring.DistanceBasedScoreFunction.score_triple"]], "sharding (besskge.scoring.distancebasedscorefunction attribute)": [[51, "besskge.scoring.DistanceBasedScoreFunction.sharding"]], "update_sharding() (besskge.scoring.distancebasedscorefunction method)": [[51, "besskge.scoring.DistanceBasedScoreFunction.update_sharding"]], "interht (class in besskge.scoring)": [[52, "besskge.scoring.InterHT"]], "broadcasted_distance() (besskge.scoring.interht method)": [[52, "besskge.scoring.InterHT.broadcasted_distance"]], "entity_embedding (besskge.scoring.interht attribute)": [[52, "besskge.scoring.InterHT.entity_embedding"]], "forward() (besskge.scoring.interht method)": [[52, "besskge.scoring.InterHT.forward"]], "negative_sample_sharing (besskge.scoring.interht attribute)": [[52, "besskge.scoring.InterHT.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.interht method)": [[52, "besskge.scoring.InterHT.reduce_embedding"]], "relation_embedding (besskge.scoring.interht attribute)": [[52, "besskge.scoring.InterHT.relation_embedding"]], "score_heads() (besskge.scoring.interht method)": [[52, "besskge.scoring.InterHT.score_heads"]], "score_tails() (besskge.scoring.interht method)": [[52, "besskge.scoring.InterHT.score_tails"]], "score_triple() (besskge.scoring.interht method)": [[52, "besskge.scoring.InterHT.score_triple"]], "sharding (besskge.scoring.interht attribute)": [[52, "besskge.scoring.InterHT.sharding"]], "update_sharding() (besskge.scoring.interht method)": [[52, "besskge.scoring.InterHT.update_sharding"]], "matrixdecompositionscorefunction (class in besskge.scoring)": [[53, "besskge.scoring.MatrixDecompositionScoreFunction"]], "broadcasted_dot_product() (besskge.scoring.matrixdecompositionscorefunction method)": [[53, "besskge.scoring.MatrixDecompositionScoreFunction.broadcasted_dot_product"]], "entity_embedding (besskge.scoring.matrixdecompositionscorefunction attribute)": [[53, "besskge.scoring.MatrixDecompositionScoreFunction.entity_embedding"]], "forward() (besskge.scoring.matrixdecompositionscorefunction method)": [[53, "besskge.scoring.MatrixDecompositionScoreFunction.forward"]], "negative_sample_sharing (besskge.scoring.matrixdecompositionscorefunction attribute)": [[53, "besskge.scoring.MatrixDecompositionScoreFunction.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.matrixdecompositionscorefunction method)": [[53, "besskge.scoring.MatrixDecompositionScoreFunction.reduce_embedding"]], "relation_embedding (besskge.scoring.matrixdecompositionscorefunction attribute)": [[53, "besskge.scoring.MatrixDecompositionScoreFunction.relation_embedding"]], "score_heads() (besskge.scoring.matrixdecompositionscorefunction method)": [[53, "besskge.scoring.MatrixDecompositionScoreFunction.score_heads"]], "score_tails() (besskge.scoring.matrixdecompositionscorefunction method)": [[53, "besskge.scoring.MatrixDecompositionScoreFunction.score_tails"]], "score_triple() (besskge.scoring.matrixdecompositionscorefunction method)": [[53, "besskge.scoring.MatrixDecompositionScoreFunction.score_triple"]], "sharding (besskge.scoring.matrixdecompositionscorefunction attribute)": [[53, "besskge.scoring.MatrixDecompositionScoreFunction.sharding"]], "update_sharding() (besskge.scoring.matrixdecompositionscorefunction method)": [[53, "besskge.scoring.MatrixDecompositionScoreFunction.update_sharding"]], "pairre (class in besskge.scoring)": [[54, "besskge.scoring.PairRE"]], "broadcasted_distance() (besskge.scoring.pairre method)": [[54, "besskge.scoring.PairRE.broadcasted_distance"]], "entity_embedding (besskge.scoring.pairre attribute)": [[54, "besskge.scoring.PairRE.entity_embedding"]], "forward() (besskge.scoring.pairre method)": [[54, "besskge.scoring.PairRE.forward"]], "negative_sample_sharing (besskge.scoring.pairre attribute)": [[54, "besskge.scoring.PairRE.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.pairre method)": [[54, "besskge.scoring.PairRE.reduce_embedding"]], "relation_embedding (besskge.scoring.pairre attribute)": [[54, "besskge.scoring.PairRE.relation_embedding"]], "score_heads() (besskge.scoring.pairre method)": [[54, "besskge.scoring.PairRE.score_heads"]], "score_tails() (besskge.scoring.pairre method)": [[54, "besskge.scoring.PairRE.score_tails"]], "score_triple() (besskge.scoring.pairre method)": [[54, "besskge.scoring.PairRE.score_triple"]], "sharding (besskge.scoring.pairre attribute)": [[54, "besskge.scoring.PairRE.sharding"]], "update_sharding() (besskge.scoring.pairre method)": [[54, "besskge.scoring.PairRE.update_sharding"]], "rotate (class in besskge.scoring)": [[55, "besskge.scoring.RotatE"]], "broadcasted_distance() (besskge.scoring.rotate method)": [[55, "besskge.scoring.RotatE.broadcasted_distance"]], "entity_embedding (besskge.scoring.rotate attribute)": [[55, "besskge.scoring.RotatE.entity_embedding"]], "forward() (besskge.scoring.rotate method)": [[55, "besskge.scoring.RotatE.forward"]], "negative_sample_sharing (besskge.scoring.rotate attribute)": [[55, "besskge.scoring.RotatE.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.rotate method)": [[55, "besskge.scoring.RotatE.reduce_embedding"]], "relation_embedding (besskge.scoring.rotate attribute)": [[55, "besskge.scoring.RotatE.relation_embedding"]], "score_heads() (besskge.scoring.rotate method)": [[55, "besskge.scoring.RotatE.score_heads"]], "score_tails() (besskge.scoring.rotate method)": [[55, "besskge.scoring.RotatE.score_tails"]], "score_triple() (besskge.scoring.rotate method)": [[55, "besskge.scoring.RotatE.score_triple"]], "sharding (besskge.scoring.rotate attribute)": [[55, "besskge.scoring.RotatE.sharding"]], "update_sharding() (besskge.scoring.rotate method)": [[55, "besskge.scoring.RotatE.update_sharding"]], "trans (class in besskge.scoring)": [[56, "besskge.scoring.TranS"]], "broadcasted_distance() (besskge.scoring.trans method)": [[56, "besskge.scoring.TranS.broadcasted_distance"]], "entity_embedding (besskge.scoring.trans attribute)": [[56, "besskge.scoring.TranS.entity_embedding"]], "forward() (besskge.scoring.trans method)": [[56, "besskge.scoring.TranS.forward"]], "negative_sample_sharing (besskge.scoring.trans attribute)": [[56, "besskge.scoring.TranS.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.trans method)": [[56, "besskge.scoring.TranS.reduce_embedding"]], "relation_embedding (besskge.scoring.trans attribute)": [[56, "besskge.scoring.TranS.relation_embedding"]], "score_heads() (besskge.scoring.trans method)": [[56, "besskge.scoring.TranS.score_heads"]], "score_tails() (besskge.scoring.trans method)": [[56, "besskge.scoring.TranS.score_tails"]], "score_triple() (besskge.scoring.trans method)": [[56, "besskge.scoring.TranS.score_triple"]], "sharding (besskge.scoring.trans attribute)": [[56, "besskge.scoring.TranS.sharding"]], "update_sharding() (besskge.scoring.trans method)": [[56, "besskge.scoring.TranS.update_sharding"]], "transe (class in besskge.scoring)": [[57, "besskge.scoring.TransE"]], "broadcasted_distance() (besskge.scoring.transe method)": [[57, "besskge.scoring.TransE.broadcasted_distance"]], "entity_embedding (besskge.scoring.transe attribute)": [[57, "besskge.scoring.TransE.entity_embedding"]], "forward() (besskge.scoring.transe method)": [[57, "besskge.scoring.TransE.forward"]], "negative_sample_sharing (besskge.scoring.transe attribute)": [[57, "besskge.scoring.TransE.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.transe method)": [[57, "besskge.scoring.TransE.reduce_embedding"]], "relation_embedding (besskge.scoring.transe attribute)": [[57, "besskge.scoring.TransE.relation_embedding"]], "score_heads() (besskge.scoring.transe method)": [[57, "besskge.scoring.TransE.score_heads"]], "score_tails() (besskge.scoring.transe method)": [[57, "besskge.scoring.TransE.score_tails"]], "score_triple() (besskge.scoring.transe method)": [[57, "besskge.scoring.TransE.score_triple"]], "sharding (besskge.scoring.transe attribute)": [[57, "besskge.scoring.TransE.sharding"]], "update_sharding() (besskge.scoring.transe method)": [[57, "besskge.scoring.TransE.update_sharding"]], "triplere (class in besskge.scoring)": [[58, "besskge.scoring.TripleRE"]], "broadcasted_distance() (besskge.scoring.triplere method)": [[58, "besskge.scoring.TripleRE.broadcasted_distance"]], "entity_embedding (besskge.scoring.triplere attribute)": [[58, "besskge.scoring.TripleRE.entity_embedding"]], "forward() (besskge.scoring.triplere method)": [[58, "besskge.scoring.TripleRE.forward"]], "negative_sample_sharing (besskge.scoring.triplere attribute)": [[58, "besskge.scoring.TripleRE.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.triplere method)": [[58, "besskge.scoring.TripleRE.reduce_embedding"]], "relation_embedding (besskge.scoring.triplere attribute)": [[58, "besskge.scoring.TripleRE.relation_embedding"]], "score_heads() (besskge.scoring.triplere method)": [[58, "besskge.scoring.TripleRE.score_heads"]], "score_tails() (besskge.scoring.triplere method)": [[58, "besskge.scoring.TripleRE.score_tails"]], "score_triple() (besskge.scoring.triplere method)": [[58, "besskge.scoring.TripleRE.score_triple"]], "sharding (besskge.scoring.triplere attribute)": [[58, "besskge.scoring.TripleRE.sharding"]], "update_sharding() (besskge.scoring.triplere method)": [[58, "besskge.scoring.TripleRE.update_sharding"]], "besskge.sharding": [[59, "module-besskge.sharding"]], "partitionedtripleset (class in besskge.sharding)": [[60, "besskge.sharding.PartitionedTripleSet"]], "create_from_dataset() (besskge.sharding.partitionedtripleset class method)": [[60, "besskge.sharding.PartitionedTripleSet.create_from_dataset"]], "create_from_queries() (besskge.sharding.partitionedtripleset class method)": [[60, "besskge.sharding.PartitionedTripleSet.create_from_queries"]], "dummy (besskge.sharding.partitionedtripleset attribute)": [[60, "besskge.sharding.PartitionedTripleSet.dummy"]], "inverse_triples (besskge.sharding.partitionedtripleset attribute)": [[60, "besskge.sharding.PartitionedTripleSet.inverse_triples"]], "neg_heads (besskge.sharding.partitionedtripleset attribute)": [[60, "besskge.sharding.PartitionedTripleSet.neg_heads"]], "neg_tails (besskge.sharding.partitionedtripleset attribute)": [[60, "besskge.sharding.PartitionedTripleSet.neg_tails"]], "partition_mode (besskge.sharding.partitionedtripleset attribute)": [[60, "besskge.sharding.PartitionedTripleSet.partition_mode"]], "sharding (besskge.sharding.partitionedtripleset attribute)": [[60, "besskge.sharding.PartitionedTripleSet.sharding"]], "triple_counts (besskge.sharding.partitionedtripleset attribute)": [[60, "besskge.sharding.PartitionedTripleSet.triple_counts"]], "triple_offsets (besskge.sharding.partitionedtripleset attribute)": [[60, "besskge.sharding.PartitionedTripleSet.triple_offsets"]], "triple_sort_idx (besskge.sharding.partitionedtripleset attribute)": [[60, "besskge.sharding.PartitionedTripleSet.triple_sort_idx"]], "triples (besskge.sharding.partitionedtripleset attribute)": [[60, "besskge.sharding.PartitionedTripleSet.triples"]], "types (besskge.sharding.partitionedtripleset attribute)": [[60, "besskge.sharding.PartitionedTripleSet.types"]], "sharding (class in besskge.sharding)": [[61, "besskge.sharding.Sharding"]], "create() (besskge.sharding.sharding class method)": [[61, "besskge.sharding.Sharding.create"]], "entity_to_idx (besskge.sharding.sharding attribute)": [[61, "besskge.sharding.Sharding.entity_to_idx"]], "entity_to_shard (besskge.sharding.sharding attribute)": [[61, "besskge.sharding.Sharding.entity_to_shard"]], "entity_type_counts (besskge.sharding.sharding attribute)": [[61, "besskge.sharding.Sharding.entity_type_counts"]], "entity_type_offsets (besskge.sharding.sharding attribute)": [[61, "besskge.sharding.Sharding.entity_type_offsets"]], "load() (besskge.sharding.sharding class method)": [[61, "besskge.sharding.Sharding.load"]], "max_entity_per_shard (besskge.sharding.sharding property)": [[61, "besskge.sharding.Sharding.max_entity_per_shard"]], "n_entity (besskge.sharding.sharding property)": [[61, "besskge.sharding.Sharding.n_entity"]], "n_shard (besskge.sharding.sharding attribute)": [[61, "besskge.sharding.Sharding.n_shard"]], "save() (besskge.sharding.sharding method)": [[61, "besskge.sharding.Sharding.save"]], "shard_and_idx_to_entity (besskge.sharding.sharding attribute)": [[61, "besskge.sharding.Sharding.shard_and_idx_to_entity"]], "shard_counts (besskge.sharding.sharding attribute)": [[61, "besskge.sharding.Sharding.shard_counts"]], "besskge.utils": [[62, "module-besskge.utils"]], "complex_multiplication() (in module besskge.utils)": [[63, "besskge.utils.complex_multiplication"]], "complex_rotation() (in module besskge.utils)": [[64, "besskge.utils.complex_rotation"]], "gather_indices() (in module besskge.utils)": [[65, "besskge.utils.gather_indices"]], "get_entity_filter() (in module besskge.utils)": [[66, "besskge.utils.get_entity_filter"]], "besskge": [[67, "module-besskge"]]}})
\ No newline at end of file
+Search.setIndex({"docnames": ["API_reference", "bess", "bibliography", "contrib", "dev_guide", "generated/besskge.batch_sampler", "generated/besskge.batch_sampler.RandomShardedBatchSampler", "generated/besskge.batch_sampler.RigidShardedBatchSampler", "generated/besskge.batch_sampler.ShardedBatchSampler", "generated/besskge.bess", "generated/besskge.bess.AllScoresBESS", "generated/besskge.bess.BessKGE", "generated/besskge.bess.EmbeddingMovingBessKGE", "generated/besskge.bess.ScoreMovingBessKGE", "generated/besskge.bess.TopKQueryBessKGE", "generated/besskge.dataset", "generated/besskge.dataset.KGDataset", "generated/besskge.embedding", "generated/besskge.embedding.init_KGE_normal", "generated/besskge.embedding.init_KGE_uniform", "generated/besskge.embedding.init_uniform_norm", "generated/besskge.embedding.init_uniform_rotation", "generated/besskge.embedding.init_xavier_norm", "generated/besskge.embedding.initialize_entity_embedding", "generated/besskge.embedding.initialize_relation_embedding", "generated/besskge.embedding.refactor_embedding_sharding", "generated/besskge.loss", "generated/besskge.loss.BaseLossFunction", "generated/besskge.loss.LogSigmoidLoss", "generated/besskge.loss.MarginBasedLossFunction", "generated/besskge.loss.MarginRankingLoss", "generated/besskge.loss.SampledSoftmaxCrossEntropyLoss", "generated/besskge.metric", "generated/besskge.metric.BaseMetric", "generated/besskge.metric.Evaluation", "generated/besskge.metric.HitsAtK", "generated/besskge.metric.METRICS_DICT", "generated/besskge.metric.ReciprocalRank", "generated/besskge.negative_sampler", "generated/besskge.negative_sampler.PlaceholderNegativeSampler", "generated/besskge.negative_sampler.RandomShardedNegativeSampler", "generated/besskge.negative_sampler.ShardedNegativeSampler", "generated/besskge.negative_sampler.TripleBasedShardedNegativeSampler", "generated/besskge.negative_sampler.TypeBasedShardedNegativeSampler", "generated/besskge.pipeline", "generated/besskge.pipeline.AllScoresPipeline", "generated/besskge.scoring", "generated/besskge.scoring.BaseScoreFunction", "generated/besskge.scoring.BoxE", "generated/besskge.scoring.ComplEx", "generated/besskge.scoring.ConvE", "generated/besskge.scoring.DistMult", "generated/besskge.scoring.DistanceBasedScoreFunction", "generated/besskge.scoring.InterHT", "generated/besskge.scoring.MatrixDecompositionScoreFunction", "generated/besskge.scoring.PairRE", "generated/besskge.scoring.RotatE", "generated/besskge.scoring.TranS", "generated/besskge.scoring.TransE", "generated/besskge.scoring.TripleRE", "generated/besskge.sharding", "generated/besskge.sharding.PartitionedTripleSet", "generated/besskge.sharding.Sharding", "generated/besskge.utils", "generated/besskge.utils.complex_multiplication", "generated/besskge.utils.complex_rotation", "generated/besskge.utils.gather_indices", "generated/besskge.utils.get_entity_filter", "index", "user_guide"], "filenames": ["API_reference.rst", "bess.rst", "bibliography.rst", "contrib.md", "dev_guide.rst", "generated/besskge.batch_sampler.rst", "generated/besskge.batch_sampler.RandomShardedBatchSampler.rst", "generated/besskge.batch_sampler.RigidShardedBatchSampler.rst", "generated/besskge.batch_sampler.ShardedBatchSampler.rst", "generated/besskge.bess.rst", "generated/besskge.bess.AllScoresBESS.rst", "generated/besskge.bess.BessKGE.rst", "generated/besskge.bess.EmbeddingMovingBessKGE.rst", "generated/besskge.bess.ScoreMovingBessKGE.rst", "generated/besskge.bess.TopKQueryBessKGE.rst", "generated/besskge.dataset.rst", "generated/besskge.dataset.KGDataset.rst", "generated/besskge.embedding.rst", "generated/besskge.embedding.init_KGE_normal.rst", "generated/besskge.embedding.init_KGE_uniform.rst", "generated/besskge.embedding.init_uniform_norm.rst", "generated/besskge.embedding.init_uniform_rotation.rst", "generated/besskge.embedding.init_xavier_norm.rst", "generated/besskge.embedding.initialize_entity_embedding.rst", "generated/besskge.embedding.initialize_relation_embedding.rst", "generated/besskge.embedding.refactor_embedding_sharding.rst", "generated/besskge.loss.rst", "generated/besskge.loss.BaseLossFunction.rst", "generated/besskge.loss.LogSigmoidLoss.rst", "generated/besskge.loss.MarginBasedLossFunction.rst", "generated/besskge.loss.MarginRankingLoss.rst", "generated/besskge.loss.SampledSoftmaxCrossEntropyLoss.rst", "generated/besskge.metric.rst", "generated/besskge.metric.BaseMetric.rst", "generated/besskge.metric.Evaluation.rst", "generated/besskge.metric.HitsAtK.rst", "generated/besskge.metric.METRICS_DICT.rst", "generated/besskge.metric.ReciprocalRank.rst", "generated/besskge.negative_sampler.rst", "generated/besskge.negative_sampler.PlaceholderNegativeSampler.rst", "generated/besskge.negative_sampler.RandomShardedNegativeSampler.rst", "generated/besskge.negative_sampler.ShardedNegativeSampler.rst", "generated/besskge.negative_sampler.TripleBasedShardedNegativeSampler.rst", "generated/besskge.negative_sampler.TypeBasedShardedNegativeSampler.rst", "generated/besskge.pipeline.rst", "generated/besskge.pipeline.AllScoresPipeline.rst", "generated/besskge.scoring.rst", "generated/besskge.scoring.BaseScoreFunction.rst", "generated/besskge.scoring.BoxE.rst", "generated/besskge.scoring.ComplEx.rst", "generated/besskge.scoring.ConvE.rst", "generated/besskge.scoring.DistMult.rst", "generated/besskge.scoring.DistanceBasedScoreFunction.rst", "generated/besskge.scoring.InterHT.rst", "generated/besskge.scoring.MatrixDecompositionScoreFunction.rst", "generated/besskge.scoring.PairRE.rst", "generated/besskge.scoring.RotatE.rst", "generated/besskge.scoring.TranS.rst", "generated/besskge.scoring.TransE.rst", "generated/besskge.scoring.TripleRE.rst", "generated/besskge.sharding.rst", "generated/besskge.sharding.PartitionedTripleSet.rst", "generated/besskge.sharding.Sharding.rst", "generated/besskge.utils.rst", "generated/besskge.utils.complex_multiplication.rst", "generated/besskge.utils.complex_rotation.rst", "generated/besskge.utils.gather_indices.rst", "generated/besskge.utils.get_entity_filter.rst", "index.rst", "user_guide.rst"], "titles": ["BESS-KGE API Reference", "BESS overview", "Bibliography", "How to contribute to the BESS-KGE project", "How to contribute to the BESS-KGE project", "besskge.batch_sampler", "besskge.batch_sampler.RandomShardedBatchSampler", "besskge.batch_sampler.RigidShardedBatchSampler", "besskge.batch_sampler.ShardedBatchSampler", "besskge.bess", "besskge.bess.AllScoresBESS", "besskge.bess.BessKGE", "besskge.bess.EmbeddingMovingBessKGE", "besskge.bess.ScoreMovingBessKGE", "besskge.bess.TopKQueryBessKGE", "besskge.dataset", "besskge.dataset.KGDataset", "besskge.embedding", "besskge.embedding.init_KGE_normal", "besskge.embedding.init_KGE_uniform", "besskge.embedding.init_uniform_norm", "besskge.embedding.init_uniform_rotation", "besskge.embedding.init_xavier_norm", "besskge.embedding.initialize_entity_embedding", "besskge.embedding.initialize_relation_embedding", "besskge.embedding.refactor_embedding_sharding", "besskge.loss", "besskge.loss.BaseLossFunction", "besskge.loss.LogSigmoidLoss", "besskge.loss.MarginBasedLossFunction", "besskge.loss.MarginRankingLoss", "besskge.loss.SampledSoftmaxCrossEntropyLoss", "besskge.metric", "besskge.metric.BaseMetric", "besskge.metric.Evaluation", "besskge.metric.HitsAtK", "besskge.metric.METRICS_DICT", "besskge.metric.ReciprocalRank", "besskge.negative_sampler", "besskge.negative_sampler.PlaceholderNegativeSampler", "besskge.negative_sampler.RandomShardedNegativeSampler", "besskge.negative_sampler.ShardedNegativeSampler", "besskge.negative_sampler.TripleBasedShardedNegativeSampler", "besskge.negative_sampler.TypeBasedShardedNegativeSampler", "besskge.pipeline", "besskge.pipeline.AllScoresPipeline", "besskge.scoring", "besskge.scoring.BaseScoreFunction", "besskge.scoring.BoxE", "besskge.scoring.ComplEx", "besskge.scoring.ConvE", "besskge.scoring.DistMult", "besskge.scoring.DistanceBasedScoreFunction", "besskge.scoring.InterHT", "besskge.scoring.MatrixDecompositionScoreFunction", "besskge.scoring.PairRE", "besskge.scoring.RotatE", "besskge.scoring.TranS", "besskge.scoring.TransE", "besskge.scoring.TripleRE", "besskge.sharding", "besskge.sharding.PartitionedTripleSet", "besskge.sharding.Sharding", "besskge.utils", "besskge.utils.complex_multiplication", "besskge.utils.complex_rotation", "besskge.utils.gather_indices", "besskge.utils.get_entity_filter", "BESS-KGE", "User guide"], "terms": {"when": [1, 3, 4, 11, 12, 13, 14, 24, 27, 28, 29, 30, 31, 38, 39, 42], "distribut": [1, 5, 9, 10, 11, 14, 18, 19, 20, 21, 60, 68, 69], "workload": 1, "over": [1, 6, 7, 8, 14, 68], "n": [1, 42, 61], "worker": [1, 6, 7, 8, 68], "ipu": [1, 3, 4, 6, 7, 8, 9, 10, 45, 66, 68, 69], "randomli": [1, 16], "split": [1, 16, 42], "entiti": [1, 2, 6, 7, 8, 10, 11, 12, 13, 14, 16, 17, 23, 25, 31, 34, 35, 38, 39, 40, 41, 42, 43, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61, 62, 67, 68, 69], "embed": [1, 2, 11, 12, 13, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 68, 69], "tabl": [1, 11, 12, 13, 17, 21, 23, 24, 25, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 69], "shard": [1, 6, 7, 8, 10, 12, 13, 14, 23, 25, 39, 40, 41, 42, 43, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 68], "equal": [1, 12], "size": [1, 6, 7, 8, 10, 13, 14, 42, 45, 48, 49, 50, 51, 53, 55, 56, 57, 58, 59, 69], "each": [1, 5, 6, 7, 8, 10, 11, 12, 13, 14, 16, 20, 21, 23, 24, 34, 40, 42, 45, 46, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61, 62, 67], "which": [1, 12, 16, 39, 40, 41, 42, 43, 67], "i": [1, 3, 4, 6, 7, 8, 12, 13, 14, 16, 23, 24, 35, 40, 42, 45, 48, 50, 61, 65, 66, 67, 68], "store": [1, 13, 15, 16, 45, 61, 68, 69], "": [1, 2, 3, 4], "memori": [1, 6, 7, 8, 11, 12, 13, 68], "The": [1, 3, 4, 6, 7, 8, 10, 11, 12, 13, 16, 25, 27, 28, 29, 30, 31, 34, 42, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61, 62, 67, 68, 69], "relat": [1, 2, 10, 11, 12, 13, 14, 16, 17, 21, 24, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61, 68], "type": [1, 6, 7, 8, 10, 11, 12, 13, 14, 16, 18, 19, 20, 21, 22, 23, 24, 25, 27, 28, 29, 30, 31, 34, 42, 43, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61, 62, 64, 65, 66, 67, 68], "other": [1, 6, 7, 8, 11, 12, 13], "hand": 1, "replic": [1, 13], "across": [1, 3, 4], "usual": 1, "much": 1, "smaller": 1, "figur": 1, "1": [1, 2, 10, 11, 12, 13, 14, 16, 18, 19, 20, 22, 28, 29, 30, 31, 34, 42, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61, 66, 69], "3": [1, 3, 4, 16, 50, 67, 69], "induc": 1, "partit": [1, 6, 7, 8, 10, 11, 14, 39, 40, 41, 42, 43, 45, 61], "tripl": [1, 2, 5, 6, 7, 8, 10, 11, 12, 13, 14, 15, 16, 24, 27, 28, 29, 30, 31, 34, 35, 39, 40, 41, 42, 43, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 67, 68], "dataset": [1, 2, 24, 50, 61, 68, 69], "accord": [1, 5, 11, 12, 13, 18, 19, 20, 22, 25, 34], "pair": [1, 2, 40, 61, 67], "head": [1, 2, 10, 11, 12, 13, 14, 16, 38, 40, 42, 43, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61, 67], "tail": [1, 2, 10, 11, 12, 13, 14, 16, 38, 40, 42, 43, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61, 67], "At": [1, 7, 14], "execut": [1, 60], "time": [1, 6, 7, 8, 68], "both": [1, 3, 4, 27, 47, 68], "train": [1, 9, 11, 12, 13, 16, 44, 45, 68, 69], "infer": [1, 2, 6, 7, 8, 9, 10, 11, 14, 44, 45, 68, 69], "batch": [1, 5, 6, 7, 8, 10, 11, 12, 13, 14, 26, 27, 28, 29, 30, 31, 34, 40, 42, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 64, 65, 68, 69], "ar": [1, 3, 4, 10, 12, 13, 14, 16, 24, 27, 34, 42, 48, 50, 61, 66, 68, 69], "construct": [1, 27, 28, 29, 30, 31, 38, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61, 62, 67], "sampl": [1, 2, 5, 6, 7, 8, 11, 12, 13, 14, 26, 27, 28, 29, 30, 31, 38, 39, 40, 41, 42, 43, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 68, 69], "uniformli": [1, 21], "from": [1, 2, 3, 4, 6, 7, 8, 11, 12, 13, 16, 25, 34, 39, 40, 41, 42, 43, 50, 61, 66], "2": [1, 11, 12, 13, 21, 43, 48, 50, 61, 64, 65, 66, 67, 69], "neg": [1, 5, 6, 7, 8, 10, 11, 12, 13, 14, 16, 26, 27, 28, 29, 30, 31, 38, 39, 40, 41, 42, 43, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61, 69], "us": [1, 2, 3, 4, 6, 7, 8, 10, 11, 13, 14, 16, 21, 27, 28, 29, 30, 31, 34, 38, 39, 40, 42, 45, 50, 52, 61, 68, 69], "corrupt": [1, 6, 7, 8, 10, 11, 12, 13, 34, 38, 39, 40, 41, 42, 43, 50], "order": [1, 14, 16, 34, 42, 45, 61, 69], "also": [1, 3, 4, 62, 69], "balanc": [1, 2, 62, 68], "wai": [1, 16], "ensur": 1, "varieti": 1, "benefici": 1, "final": [1, 3, 4, 14, 48, 50], "qualiti": [1, 16], "left": [1, 3, 4], "A": [1, 2, 34, 39, 61, 62], "made": 1, "up": [1, 3, 4, 69], "9": [1, 69], "block": [1, 10], "contain": [1, 3, 4, 16, 61], "same": [1, 7, 10, 14, 16, 23, 24, 42, 43, 61, 62, 66, 67], "number": [1, 6, 7, 8, 10, 11, 12, 13, 14, 16, 23, 24, 31, 40, 42, 45, 48, 49, 50, 51, 53, 55, 56, 57, 58, 59, 61, 62, 69], "j": [1, 67], "0": [1, 3, 4, 6, 7, 8, 16, 18, 19, 21, 22, 28, 29, 30, 31, 34, 39, 48, 50, 53, 57, 59, 61, 66, 69], "right": 1, "all": [1, 3, 4, 6, 7, 10, 13, 14, 16, 34, 39, 42, 43, 45, 50, 61, 66], "possibli": [1, 16, 42, 45, 50, 61], "pad": [1, 6, 7, 11, 12, 13, 14, 42, 62], "In": [1, 2, 3, 4], "thi": [1, 3, 4, 6, 7, 8, 12, 13, 14, 16, 23, 24, 45, 50, 61, 68], "exampl": [1, 13, 14, 37], "scheme": [1, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 22, 68], "allow": [1, 3, 4, 68], "u": [1, 59], "commun": [1, 68], "first": [1, 3, 4, 16, 40, 67], "need": [1, 16, 23, 24, 27, 28, 29, 30, 31, 45, 48, 50, 68], "gather": [1, 10, 11, 12, 13, 14, 42, 66, 68], "its": [1, 24], "chip": [1, 68], "posit": [1, 5, 6, 7, 8, 11, 12, 13, 19, 26, 27, 28, 29, 30, 31, 34, 46], "These": 1, "includ": [1, 50], "itself": 1, "peer": 1, "requir": [1, 3, 4, 11, 12, 13, 14, 34, 40, 66], "sram": [1, 69], "retriev": [1, 68], "triangl": 1, "colour": 1, "addit": [1, 3, 4, 24], "portion": 1, "can": [1, 3, 4, 12, 13, 14, 23, 24, 34, 45, 61, 68], "reconstruct": 1, "share": [1, 2, 6, 7, 8, 11, 12, 13, 27, 40, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 62, 68], "between": [1, 2, 12, 13, 21, 48, 68], "through": [1, 3, 4, 11, 12, 13, 50], "alltoal": [1, 10, 12, 13, 14], "collect": [1, 6, 7, 8, 10, 11, 12, 13, 14, 15, 61, 68], "oper": [1, 3, 4, 11, 12, 13, 50, 68], "remain": [1, 62], "place": 1, "score": [1, 10, 11, 12, 13, 14, 21, 26, 27, 28, 29, 30, 31, 34, 39, 42, 45, 61, 68], "where": [1, 3, 4, 10, 12, 13, 14, 16, 34, 35, 40, 42, 61], "4": [1, 2, 3, 4, 69], "exchang": 1, "an": [1, 3, 4, 10, 13, 14, 16, 23, 24, 45, 68, 69], "red": 1, "arrow": 1, "effect": [1, 39], "transpos": 1, "row": [1, 20, 34, 42, 64, 65, 66, 67], "column": [1, 16], "pictur": 1, "after": [1, 10, 45, 50, 62], "ha": [1, 6, 7, 8, 20, 69], "correct": [1, 13, 14], "comput": [1, 2, 10, 11, 12, 13, 14, 26, 27, 28, 29, 30, 31, 32, 34, 37, 45, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 68, 69], "present": [1, 16], "abov": [1, 3, 4], "implement": [1, 9], "besskg": [1, 3, 4, 68, 69], "embeddingmovingbesskg": [1, 13, 68], "while": [1, 14], "alwai": [1, 27], "turn": 1, "out": [1, 45], "expens": 1, "mani": 1, "per": [1, 6, 7, 8, 39, 40, 41, 42, 43], "dimens": [1, 34, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "larg": [1, 2, 10, 13, 14, 45], "case": 1, "scoremovingbesskg": [1, 10, 14, 68], "increas": 1, "overal": 1, "throughput": [1, 68], "altern": 1, "work": [1, 3, 4], "well": [1, 3, 4, 53, 57], "differ": [1, 10, 14, 23, 24, 25, 34, 50, 68], "li": [1, 2], "how": 1, "instead": [1, 39, 40, 41, 42, 43, 45, 48], "send": [1, 13], "queri": [1, 10, 13, 14, 34, 39, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61], "devic": [1, 5, 10, 11, 12, 13, 14, 39, 40, 41, 42, 43], "allgath": [1, 13], "against": [1, 10, 12, 13, 14, 39, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61], "partial": 1, "set": [1, 3, 4, 10, 11, 14, 16, 24, 34, 42, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 67, 69], "sent": 1, "via": [1, 2, 10, 14], "new": [1, 3, 4, 6, 7, 8, 25, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 68], "cheaper": 1, "although": 1, "5": 1, "correspond": [1, 13, 23, 24, 42, 66], "6": [1, 2, 3, 4, 48], "put": 1, "back": [1, 13, 62], "came": 1, "complet": [1, 2, 10, 14, 16, 45, 61], "respons": [1, 14], "acls20": [2, 48], "ralph": 2, "abboud": 2, "i\u0307smail": 2, "i\u0307lkan": 2, "ceylan": 2, "thoma": 2, "lukasiewicz": 2, "tommaso": 2, "salvatori": 2, "box": [2, 68], "model": [2, 6, 7, 8, 11, 32, 45, 46, 48, 49, 50, 51, 53, 55, 56, 57, 58, 59, 68], "knowledg": [2, 10, 13, 14, 15, 16, 31, 39, 48, 49, 50, 51, 53, 55, 56, 57, 58, 59, 61, 62, 68, 69], "base": [2, 6, 7, 8, 10, 11, 14, 16, 26, 27, 28, 29, 30, 31, 41, 42, 43, 45, 47, 48, 52, 54, 61], "advanc": 2, "neural": 2, "inform": 2, "process": [2, 5, 6, 7, 8, 39, 40, 41, 42, 43], "system": [2, 3, 4, 69], "33": 2, "annual": 2, "confer": 2, "2020": 2, "neurip": 2, "13": [2, 58], "asam20": [2, 16], "breit": 2, "anna": 2, "ott": 2, "simon": 2, "agibetov": 2, "asan": 2, "samwald": 2, "matthia": 2, "openbiolink": [2, 16], "benchmark": [2, 16], "framework": [2, 11, 68], "scale": [2, 22, 27, 28, 29, 30, 31], "biomed": 2, "link": [2, 3, 4, 34, 68, 69], "predict": [2, 14, 32, 34, 35, 37, 45, 69], "bioinformat": 2, "36": 2, "4097": 2, "4098": 2, "bugd": [2, 58], "antoin": 2, "bord": 2, "nicola": 2, "usuni": 2, "alberto": 2, "garc\u00eda": 2, "dur\u00e1n": 2, "jason": 2, "weston": 2, "oksana": 2, "yakhnenko": 2, "translat": 2, "multi": 2, "data": [2, 6, 7, 8, 10, 16], "26": 2, "27th": 2, "2013": 2, "2787": 2, "2795": 2, "cjm": [2, 9, 10, 11, 14, 31], "22": [2, 9, 10, 11, 14, 31, 53, 59], "cattaneo": 2, "daniel": 2, "justu": 2, "harri": 2, "mellor": 2, "dougla": 2, "orr": 2, "jerom": 2, "maloberti": 2, "zheni": 2, "liu": 2, "thorin": 2, "farnsworth": 2, "andrew": 2, "fitzgibbon": 2, "blazej": 2, "banaszewski": 2, "carlo": 2, "luschi": 2, "bess": [2, 5, 39, 42, 44, 69], "graph": [2, 10, 13, 14, 15, 16, 31, 39, 48, 49, 50, 51, 53, 55, 56, 57, 58, 59, 61, 62, 68, 69], "arxiv": 2, "preprint": 2, "2211": 2, "12281": 2, "2022": 2, "chwc21": [2, 55], "linlin": 2, "chao": 2, "jianshan": 2, "he": 2, "taifeng": 2, "wang": 2, "wei": 2, "chu": 2, "pairr": [2, 68], "vector": 2, "proceed": 2, "59th": 2, "meet": 2, "associ": [2, 16], "linguist": 2, "11th": 2, "intern": [2, 27, 47], "joint": 2, "natur": 2, "languag": 2, "acl": 2, "ijcnlp": 2, "2021": 2, "volum": 2, "long": 2, "paper": 2, "virtual": 2, "event": 2, "august": 2, "4360": 2, "4369": 2, "dppr18": [2, 16, 50], "tim": 2, "dettmer": 2, "minervini": 2, "pasqual": 2, "stenetorp": 2, "pontu": 2, "sebastian": 2, "riedel": 2, "convolut": [2, 50], "2d": [2, 50], "32th": 2, "aaai": 2, "artifici": 2, "intellig": 2, "1811": 2, "1818": 2, "2018": 2, "hfz": [2, 16], "20": [2, 3, 4, 16, 69], "weihua": 2, "hu": 2, "fei": 2, "marinka": 2, "zitnik": 2, "yuxiao": 2, "dong": 2, "hongyu": 2, "ren": 2, "bowen": 2, "michel": 2, "catasta": 2, "jure": 2, "leskovec": 2, "open": [2, 3, 4, 68], "machin": [2, 68], "learn": [2, 24, 48, 49, 50, 51, 53, 55, 56, 57, 58, 59], "jcmb15": [2, 31], "\u00e9": 2, "bastien": 2, "jean": 2, "kyunghyun": 2, "cho": 2, "roland": 2, "memisev": 2, "yoshua": 2, "bengio": 2, "On": 2, "veri": [2, 13], "target": 2, "vocabulari": 2, "53rd": 2, "7th": 2, "10": [2, 16, 45, 69], "2015": 2, "mbs15": [2, 16], "farzaneh": 2, "mahdisoltani": 2, "joanna": 2, "biega": 2, "fabian": 2, "m": [2, 3, 4, 69], "suchanek": 2, "yago3": [2, 16, 69], "multilingu": 2, "wikipedia": 2, "seventh": 2, "biennial": 2, "innov": 2, "research": [2, 3, 4, 69], "cidr": 2, "asilomar": 2, "ca": 2, "usa": 2, "januari": 2, "7": [2, 16, 69], "onlin": 2, "sdnt19": [2, 28, 56], "zhiqe": 2, "sun": 2, "zhi": 2, "hong": 2, "deng": 2, "jian": 2, "yun": 2, "nie": 2, "tang": 2, "rotat": [2, 21, 65, 68], "complex": [2, 56, 64, 65, 68], "space": [2, 48], "represent": [2, 48], "iclr": 2, "2019": 2, "twr": [2, 49], "16": [2, 6, 7, 8, 49, 69], "th": 2, "o": [2, 6, 7, 8], "trouillon": 2, "johann": 2, "welbl": 2, "ric": 2, "gaussier": 2, "guillaum": 2, "bouchard": 2, "simpl": 2, "33rd": 2, "icml": 2, "2016": 2, "48": 2, "jmlr": 2, "workshop": 2, "2071": 2, "2080": 2, "wmw": [2, 53], "baoxin": 2, "qingy": 2, "meng": 2, "ziyu": 2, "honghong": 2, "zhao": 2, "dayong": 2, "wu": 2, "wanxiang": 2, "che": 2, "shijin": 2, "zhigang": 2, "chen": 2, "cong": 2, "interht": [2, 68], "interact": 2, "2202": 2, "04897": 2, "yyh": [2, 51], "15": [2, 16, 51], "bishan": 2, "yang": 2, "wen": 2, "tau": 2, "yih": 2, "xiaodong": 2, "jianfeng": 2, "gao": 2, "3rd": 2, "yll": [2, 59], "yu": 2, "zhicong": 2, "luo": 2, "huanyong": 2, "lin": 2, "hongzhu": 2, "yafeng": 2, "tripler": [2, 68], "2209": 2, "08271": 2, "zyx22": [2, 57], "xuanyu": 2, "zhang": 2, "qing": 2, "dongliang": 2, "xu": 2, "tran": [2, 68], "transit": 2, "synthet": 2, "find": [2, 67], "emnlp": 2, "1202": 2, "1208": 2, "you": [3, 4, 69], "even": [3, 4], "don": [3, 4], "t": [3, 4, 10, 12, 14, 15, 16, 39, 40, 41, 42, 43, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61, 67, 68], "have": [3, 4, 16, 45, 50, 68], "access": [3, 4, 68], "ipumodel": [3, 4], "emul": [3, 4], "most": [3, 4, 14, 35, 45], "function": [3, 4, 6, 7, 8, 10, 11, 12, 13, 14, 17, 21, 23, 24, 26, 27, 28, 29, 30, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 63, 66, 69], "physic": [3, 4, 69], "hardwar": [3, 4, 45], "tunnel": [3, 4], "web": [3, 4], "editor": [3, 4], "desktop": [3, 4], "app": [3, 4], "minimum": [3, 4], "effort": [3, 4], "excel": [3, 4], "solut": [3, 4], "test": [3, 4, 16, 69], "directli": [3, 4], "here": [3, 4, 16, 45], "do": [3, 4], "fork": [3, 4], "repositori": [3, 4], "launch": [3, 4], "hour": [3, 4], "session": [3, 4], "free": [3, 4, 68, 69], "form": [3, 4], "http": [3, 4, 16, 69], "consol": [3, 4], "com": [3, 4, 16, 69], "github": [3, 4, 16, 68, 69], "userid": [3, 4], "reponam": [3, 4], "graphcor": [3, 4, 69], "2fpytorch": [3, 4], "3a3": [3, 4], "ubuntu": [3, 4, 69], "04": [3, 4, 69], "20230703": [3, 4], "pod4": [3, 4, 69], "repopnam": [3, 4], "address": [3, 4], "e": [3, 4, 16, 64, 65, 66, 67], "g": [3, 4], "origin": [3, 4, 16], "repo": [3, 4, 68], "start": [3, 4, 34, 68], "clone": [3, 4], "termin": [3, 4], "pane": [3, 4], "run": [3, 4, 6, 7, 8, 45, 69], "command": [3, 4], "bash": [3, 4], "gradient": [3, 4, 69], "launch_vscode_serv": [3, 4], "sh": [3, 4, 69], "name": [3, 4, 10, 12, 14, 16, 45, 61], "option": [3, 4, 6, 7, 8, 10, 11, 12, 13, 14, 16, 23, 24, 34, 42, 45, 61, 62, 69], "argument": [3, 4, 50], "defin": [3, 4, 16, 48], "remot": [3, 4], "default": [3, 4, 6, 7, 8, 10, 11, 12, 13, 14, 18, 19, 22, 30, 34, 40, 42, 45, 48, 49, 50, 51, 53, 55, 56, 57, 58, 59, 61, 62], "script": [3, 4], "download": [3, 4, 16], "instal": [3, 4, 68], "depend": [3, 4, 67, 69], "ask": [3, 4], "author": [3, 4], "account": [3, 4], "write": [3, 4], "privileg": [3, 4], "provid": [3, 4, 14, 23, 24, 34, 45], "pleas": [3, 4], "refer": [3, 4, 68], "notebook": [3, 4, 69], "detail": [3, 4, 69], "step": [3, 4, 10, 11, 12, 13, 14, 45], "connect": [3, 4], "onc": [3, 4, 23, 24, 68], "dev": [3, 4], "build": [3, 4, 15, 16], "custom": [3, 4], "op": [3, 4], "now": [3, 4], "readi": [3, 4], "close": [3, 4], "stop": [3, 4], "rememb": [3, 4], "unregist": [3, 4], "explain": [3, 4], "common": [3, 4], "issu": [3, 4, 68], "paragraph": [3, 4], "To": [3, 4, 10, 11, 14, 21, 45], "resum": [3, 4], "your": [3, 4, 69], "just": [3, 4], "section": [3, 4], "profil": [3, 4], "repeat": [3, 4, 7], "chang": [3, 4, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "extens": [3, 4], "persist": [3, 4], "poplar": [3, 4, 69], "sdk": [3, 4, 69], "follow": [3, 4, 69], "instruct": [3, 4, 69], "get": [3, 4, 68], "guid": [3, 4, 68], "Then": [3, 4], "enabl": [3, 4, 69], "creat": [3, 4, 61, 62, 69], "activ": [3, 4, 30, 68, 69], "python": [3, 4, 68, 69], "virtualenv": [3, 4, 69], "poptorch": [3, 4, 6, 7, 8, 68, 69], "wheel": [3, 4, 69], "necessari": [3, 4], "python3": [3, 4, 69], "8": [3, 4, 69], "venv": [3, 4, 69], "add": [3, 4], "bin": [3, 4, 69], "sourc": [3, 4, 6, 7, 8, 10, 11, 12, 13, 14, 16, 18, 19, 20, 21, 22, 23, 24, 25, 27, 28, 29, 30, 31, 33, 34, 35, 37, 39, 40, 41, 42, 43, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61, 62, 64, 65, 66, 67, 69], "path_to_poplar_sdk": [3, 4], "pip": [3, 4, 69], "poplar_sdk_en": [3, 4, 69], "whl": [3, 4, 69], "r": [3, 4, 10, 14, 15, 16, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61, 65, 67, 68], "txt": [3, 4], "sever": [3, 4], "util": [3, 4, 6, 7, 8, 15, 17, 32, 68], "dure": [3, 4, 69], "check": [3, 4], "help": [3, 4], "list": [3, 4, 6, 7, 8, 16, 23, 24, 34, 42, 45, 48, 49, 50, 51, 53, 55, 56, 57, 58, 59], "befor": [3, 4, 11, 12, 13, 14, 48, 50, 53, 55, 57, 59], "submit": [3, 4], "pr": [3, 4], "upstream": [3, 4], "ci": [3, 4], "particular": [3, 4], "mind": [3, 4], "our": [3, 4, 69], "format": [3, 4], "error": [3, 4, 10, 14, 45, 69], "lint": [3, 4], "automat": [3, 4, 16], "insid": [3, 4, 48, 69], "unit": [3, 4], "folder": [3, 4], "individu": [3, 4], "pattern": [3, 4], "match": [3, 4], "filter": [3, 4, 11, 12, 13, 14, 45, 67], "k": [3, 4, 14, 34, 35, 36, 45, 65, 66], "cpp": [3, 4], "custom_op": [3, 4], "updat": [3, 4], "makefil": [3, 4], "ad": [3, 4, 24, 61, 68], "class": [5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62], "partitioned_triple_set": [6, 7, 8, 45], "negative_sampl": [6, 7, 8, 10, 11, 12, 13, 14, 68], "shard_b": [6, 7, 8, 10, 14], "batches_per_step": [6, 7, 8], "seed": [6, 7, 8, 16, 39, 40, 42, 43, 62], "hrt_freq_weight": [6, 7, 8], "fals": [6, 7, 8, 11, 12, 13, 14, 34, 40, 42, 45, 48, 49, 51, 53, 55, 56, 57, 58, 59, 61], "weight_smooth": [6, 7, 8], "duplicate_batch": [6, 7, 8], "return_triple_idx": [6, 7, 8], "random": [6, 16, 40, 62], "indic": [6, 7, 8, 10, 11, 12, 13, 14, 34, 42, 61, 66], "replac": 6, "No": [6, 39, 69], "appli": [6, 42, 45, 50, 53, 57, 62, 67], "initi": [6, 7, 8, 10, 11, 12, 13, 14, 17, 18, 19, 20, 21, 22, 23, 24, 27, 28, 29, 30, 31, 34, 39, 40, 42, 43, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "sampler": [6, 7, 8, 10, 11, 12, 13, 14, 39, 40, 41, 42, 43, 45, 50], "paramet": [6, 7, 8, 10, 11, 12, 13, 14, 16, 18, 19, 20, 21, 22, 23, 24, 25, 27, 28, 29, 30, 31, 34, 35, 37, 39, 40, 42, 43, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61, 62, 64, 65, 66, 67, 69], "partitionedtripleset": [6, 7, 8, 50, 68], "pre": [6, 7, 8, 16], "shardednegativesampl": [6, 7, 8, 11, 12, 13, 39, 68], "int": [6, 7, 8, 10, 11, 12, 13, 14, 16, 23, 24, 31, 35, 39, 40, 42, 43, 45, 48, 49, 50, 51, 52, 53, 55, 56, 57, 58, 59, 62], "micro": [6, 7, 8, 11, 12, 13, 14], "call": [6, 7, 8, 39], "rng": [6, 7, 8, 39, 40, 41, 42, 43], "bool": [6, 7, 8, 11, 12, 13, 14, 18, 19, 24, 27, 28, 29, 30, 31, 34, 39, 40, 41, 42, 43, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61], "If": [6, 7, 8, 11, 12, 13, 14, 16, 23, 24, 34, 40, 42, 45, 48, 49, 50, 51, 53, 55, 56, 57, 58, 59, 61, 66, 69], "true": [6, 7, 8, 11, 12, 13, 14, 18, 19, 24, 34, 40, 42, 45, 48, 49, 50, 51, 53, 55, 56, 57, 58, 59, 62], "frequenc": [6, 7, 8], "weight": [6, 7, 8, 11, 12, 13, 27, 28, 29, 30, 31, 69], "float": [6, 7, 8, 16, 18, 19, 22, 28, 29, 30, 31, 48, 50, 53, 57, 59], "smooth": [6, 7, 8], "two": [6, 7, 8, 48, 50, 67], "ident": [6, 7, 8], "halv": [6, 7, 8], "ht": [6, 7, 8, 12, 39, 40, 41, 42, 43], "return": [6, 7, 8, 10, 11, 12, 13, 14, 16, 18, 19, 20, 21, 22, 23, 24, 25, 27, 28, 29, 30, 31, 34, 35, 37, 39, 42, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61, 62, 64, 65, 66, 67], "wrt": [6, 7, 8, 16, 45], "get_dataload": [6, 7, 8], "shuffl": [6, 7, 8], "num_work": [6, 7, 8], "persistent_work": [6, 7, 8], "buffer_s": [6, 7, 8], "dataload": [6, 7, 8], "instanti": [6, 7, 8, 16], "appropri": [6, 7, 8], "iter": [6, 7, 8, 10, 14], "It": [6, 7, 8, 14, 45], "asynchron": [6, 7, 8], "load": [6, 7, 8, 14, 16, 62], "minim": [6, 7, 8, 68], "cpu": [6, 7, 8], "compil": [6, 7, 8, 69], "epoch": [6, 7, 8], "see": [6, 7, 8, 10, 11, 12, 13, 14, 28, 29, 30, 31, 34, 37, 39, 42, 43, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 68, 69], "torch": [6, 7, 8, 66], "__init__": [6, 7, 8, 30, 42, 43, 48, 49, 50, 51, 53, 55, 56, 57, 58, 59], "ring": [6, 7, 8], "buffer": [6, 7, 8], "preload": [6, 7, 8], "get_dataloader_sampl": [6, 7, 8], "sample_tripl": [6, 7, 8], "idx": [6, 7, 8], "index": [6, 7, 8, 10, 42, 66, 67], "dict": [6, 7, 8, 11, 12, 13, 14, 16, 34, 45], "str": [6, 7, 8, 11, 12, 13, 14, 16, 30, 34, 36, 39, 40, 41, 42, 43, 45, 61, 67], "union": [6, 7, 8, 14, 16, 23, 24, 45, 48, 49, 50, 51, 53, 55, 56, 57, 58, 59], "ndarrai": [6, 7, 8, 16, 42, 43, 45, 61, 62], "ani": [6, 7, 8, 11, 12, 13, 14, 16, 42, 43, 45, 61, 62, 68], "dtype": [6, 7, 8, 16, 42, 43, 45, 61, 62, 69], "int64": [6, 7, 8, 42, 61, 62], "bool_": [6, 7, 8, 42], "relev": [6, 7, 8, 11, 12, 13], "static": [6, 7, 8], "worker_init_fn": [6, 7, 8], "worker_id": [6, 7, 8], "pass": [6, 7, 8, 14, 23, 24, 25, 45, 48, 50], "id": [6, 7, 8, 16, 24, 34, 42, 43, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61, 62, 67], "none": [6, 7, 8, 10, 11, 12, 13, 14, 16, 23, 24, 34, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61, 62], "specifi": [7, 10, 16, 45, 66], "shorter": 7, "ones": [7, 24], "length": [7, 23, 24, 42], "mask": [7, 11, 12, 13, 14, 34, 42], "identifi": [7, 11, 12, 13], "abstract": [8, 11, 27, 29, 47, 52, 54], "pytorch": [9, 69], "modul": [9, 10, 11, 12, 13, 14, 27, 32, 34, 45, 47], "kge": [9, 10, 11, 12, 13, 14, 32, 46, 69], "multipl": [9, 64], "candidate_sampl": [10, 14], "score_fn": [10, 11, 12, 13, 14, 45], "window_s": [10, 14, 45], "1000": [10, 45], "h": [10, 12, 14, 15, 16, 39, 40, 41, 42, 43, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61, 67, 68], "host": [10, 11, 12, 13, 45], "combin": [10, 11, 14, 45, 50], "h_shard": [10, 14, 45, 61], "t_shard": [10, 14, 45, 61], "sinc": 10, "onli": [10, 14, 16, 39, 40, 41, 42, 43, 50, 61], "part": [10, 16, 61, 64, 65], "slide": [10, 14, 45], "window": [10, 14, 45], "metric": [10, 11, 12, 13, 14, 45, 68], "should": [10, 14, 16, 45, 50], "aggreg": 10, "pipelin": [10, 68], "allscorespipelin": [10, 68], "allscor": 10, "placeholdernegativesampl": [10, 14, 68], "basescorefunct": [10, 11, 12, 13, 14, 45, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 68], "decreas": [10, 14, 34, 45], "avoid": [10, 13, 14, 45], "oom": [10, 14, 45], "forward": [10, 11, 12, 13, 14, 27, 28, 29, 30, 31, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "similarli": [10, 14, 61], "candid": [10, 14, 34, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "thei": [10, 13, 14, 42], "togeth": [10, 14], "tensor": [10, 11, 12, 13, 14, 18, 19, 20, 21, 22, 23, 24, 27, 28, 29, 30, 31, 34, 45, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 64, 65, 66, 67], "self": [10, 27, 28, 29, 30, 31], "shape": [10, 11, 12, 13, 14, 16, 23, 25, 27, 28, 29, 30, 31, 34, 42, 43, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61, 62, 64, 65, 66, 67], "known": [10, 14, 61], "loss_fn": [11, 12, 13], "evalu": [11, 12, 13, 14, 32, 45, 68], "return_scor": [11, 12, 13, 14, 45], "augment_neg": [11, 12, 13], "ht_shardpair": [11, 61], "baselossfunct": [11, 12, 13, 28, 29, 30, 31, 68], "loss": [11, 12, 13, 50, 68], "augment": [11, 12, 13], "triple_mask": [11, 12, 13, 14, 34], "triple_weight": [11, 12, 13, 27, 28, 29, 30, 31], "negative_mask": [11, 12, 13, 14], "compris": [11, 12, 13], "four": [11, 12, 13], "phase": [11, 12, 13, 21], "local": [11, 12, 13, 16, 61, 62, 68], "n_shard": [11, 12, 13, 14, 23, 42, 61, 62], "positive_per_partit": [11, 12, 13], "b": [11, 12, 13, 14, 19, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 66], "padded_neg": [11, 12, 13, 14, 42], "discard": [11, 12, 13, 14], "properti": [11, 12, 13, 16, 62], "n_embedding_paramet": [11, 12, 13], "trainabl": [11, 12, 13], "score_batch": [11, 12, 13], "tupl": [11, 12, 13, 16, 42, 67], "n_neg": [11, 12, 13, 27, 28, 29, 30, 31, 34, 40, 42, 43, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 61], "move": [12, 13], "done": 12, "singl": [12, 34], "total": [12, 31], "disabl": 12, "otherwis": [12, 66], "conveni": 13, "so": [13, 20], "For": [13, 14, 45, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 66, 68, 69], "valu": 13, "document": 13, "multipli": [13, 53, 57], "doe": 13, "support": [13, 19, 34, 45, 69], "100": [14, 69], "specif": [14, 16, 42, 45, 46, 61], "top": [14, 34, 45], "like": [14, 34, 35, 45, 66], "input": [14, 50], "recommend": [14, 69], "one": [14, 24, 25, 42, 61], "want": [14, 16], "loop": 14, "topk": 14, "triplebasedshardednegativesampl": [14, 68], "unnecessari": 14, "best": 14, "respect": 14, "kept": 14, "next": 14, "rest": 14, "mask_on_gath": [14, 42], "n_entiti": [16, 23, 31, 62], "n_relation_typ": [16, 24, 48, 49, 50, 51, 53, 55, 56, 57, 58, 59], "original_triple_id": 16, "entity_dict": 16, "relation_dict": 16, "type_offset": [16, 62], "neg_head": [16, 61], "neg_tail": [16, 61], "repres": 16, "int32": [16, 42, 43, 45, 61, 62], "classmethod": [16, 61, 62], "build_ogbl_biokg": 16, "root": 16, "ogbl": [16, 69], "biokg": [16, 69], "ogb": 16, "stanford": 16, "edu": 16, "doc": 16, "linkprop": 16, "path": [16, 62, 69], "locat": 16, "build_ogbl_wikikg2": 16, "wikikg2": [16, 69], "build_openbiolink": 16, "high": [16, 44, 68], "version": 16, "openbiolink2020": 16, "hq": 16, "build_yago310": 16, "subgraph": 16, "least": 16, "them": [16, 42, 68, 69], "yago": 16, "org": 16, "label": 16, "from_datafram": 16, "df": 16, "head_column": 16, "relation_column": 16, "tail_column": 16, "entity_typ": 16, "1234": 16, "panda": 16, "datafram": 16, "assign": [16, 34, 61], "contigu": 16, "dictionari": [16, 34], "seri": 16, "map": [16, 62], "string": 16, "valid": 16, "instanc": 16, "from_tripl": 16, "arrai": 16, "alreadi": [16, 42], "been": [16, 50, 68, 69], "note": [16, 50], "attribut": [16, 32], "manual": 16, "numpi": 16, "head_id": 16, "relation_id": [16, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "tail_id": 16, "num_tripl": 16, "offset": [16, 53, 57, 59, 62], "ht_type": 16, "n_tripl": [16, 42, 43, 61], "h_type": [16, 61], "t_type": [16, 61], "object": [16, 62], "save": [16, 62, 69], "node": 16, "edg": 16, "n_neg_head": [16, 61], "n_neg_tail": [16, 61], "out_fil": [16, 62], "pkl": 16, "output": [16, 34, 50, 62], "file": [16, 62, 69], "h_id": 16, "r_id": 16, "t_id": 16, "assum": [16, 34, 50, 69], "cluster": [16, 42, 62], "manag": 17, "embedding_t": [18, 19, 20, 21, 22], "std": 18, "divide_by_embedding_s": [18, 19], "normal": [18, 20, 22, 48, 50, 53, 55, 57, 59], "mean": 18, "standard": [18, 22], "deviat": [18, 22], "rescal": [18, 19], "row_siz": [18, 19, 22, 23, 24, 25], "symmetr": 19, "uniform": [19, 20], "boundari": 19, "norm": [20, 48, 52, 53, 55, 56, 57, 58, 59], "entri": [21, 23, 24], "being": 21, "pi": [21, 65], "gain": 22, "xavier": 22, "fan_in": 22, "fan_out": 22, "factor": [22, 27, 28, 29, 30, 31, 59], "callabl": [23, 24, 36, 48, 49, 50, 51, 53, 55, 56, 57, 58, 59], "either": 23, "max_entity_per_shard": [23, 62], "unshard": 23, "alloc": [23, 24], "omit": [23, 24], "max_ent_per_shard": 23, "inverse_rel": [24, 48, 49, 50, 51, 53, 55, 56, 57, 58, 59], "invers": [24, 48, 49, 50, 51, 53, 55, 56, 57, 58, 59, 61], "direct": 24, "given": [24, 45, 67, 69], "entity_embed": [25, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "old_shard": 25, "new_shard": [25, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "refactor": 25, "n_shard_old": 25, "max_ent_per_shard_old": 25, "current": [25, 34], "n_shard_new": 25, "max_ent_per_shard_new": 25, "arg": [27, 47], "kwarg": [27, 47], "fp32": [27, 69], "state": [27, 47], "nn": [27, 47], "scriptmodul": [27, 47], "positive_scor": [27, 28, 29, 30, 31], "negative_scor": [27, 28, 29, 30, 31], "batch_siz": [27, 28, 29, 30, 31, 34, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "get_negative_weight": [27, 28, 29, 30, 31], "negative_adversarial_sampl": [27, 28, 29, 30, 31], "els": [27, 28, 29, 30, 31, 34, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "loss_scal": [27, 28, 29, 30, 31], "might": [27, 28, 29, 30, 31], "fp16": [27, 28, 29, 30, 31, 69], "adversari": [27, 28, 29, 30, 31], "negative_adversarial_scal": [27, 28, 29, 30, 31], "reciproc": [27, 28, 29, 30, 31, 37], "temperatur": [27, 28, 29, 30, 31], "margin": [28, 29, 30], "log": 28, "sigmoid": [28, 50], "activation_funct": 30, "relu": 30, "rank": [30, 34, 35, 37], "pairwis": 30, "hing": 30, "marginbasedlossfunct": [30, 68], "softmax": 31, "cross": 31, "entropi": 31, "metric_list": 34, "mode": [34, 61], "averag": 34, "worst_rank_infti": 34, "reduct": [34, 37, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "return_rank": 34, "mrr": [34, 36, 37], "hit": [34, 35, 36], "optimist": 34, "pessimist": 34, "infin": 34, "worst": 34, "possibl": 34, "method": 34, "reduc": [34, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "along": [34, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 66], "sum": [34, 48, 49, 50, 51, 54], "alongsid": 34, "dict_metrics_from_rank": 34, "batch_rank": 34, "element": [34, 66], "boolean": 34, "ranks_from_indic": 34, "ground_truth": [34, 61], "candidate_indic": 34, "ground": [34, 35, 37, 61], "truth": [34, 35, 37, 61], "n_candid": 34, "likelihood": 34, "distinct": 34, "among": [34, 35, 37], "ranks_from_scor": 34, "pos_scor": 34, "candidate_scor": 34, "stacked_metrics_from_rank": 34, "stack": 34, "n_metric": 34, "count": [35, 69], "maximum": [35, 69], "accept": 35, "hitsatk": [36, 68], "reciprocalrank": [36, 68], "basemetr": [37, 68], "corruption_schem": [39, 40, 41, 42, 43, 45], "placehold": 39, "topkquerybesskg": [39, 42, 68], "flat_negative_format": [39, 40, 41, 42, 43], "local_sampl": [39, 40, 41, 42, 43], "gener": [39, 40, 41, 42, 43, 63], "half": 40, "second": [40, 67, 69], "negative_head": 42, "negative_tail": 42, "return_sort_idx": 42, "predetermin": 42, "global": [42, 45, 61, 62, 67], "randomshardednegativesampl": [42, 43, 68], "sort": [42, 61], "recov": 42, "pad_neg": 42, "shard_count": [42, 62], "padded_shard_length": 42, "divid": 42, "view": 42, "shard_neg": 42, "shard_neg_count": 42, "sort_neg_idx": 42, "triple_typ": 43, "level": 44, "api": [44, 68, 69], "batch_sampl": [45, 68], "filter_tripl": [45, 67], "return_topk": 45, "use_ipu_model": 45, "kg": 45, "appear": [45, 67], "shardedbatchsampl": [45, 68], "whose": 45, "must": 45, "caus": 45, "go": 45, "actual": 45, "result": 45, "head_emb": [47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "tail_emb": [47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "score_tripl": [47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "negative_sample_shar": [47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "relation_embed": [47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "score_head": [47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "fix": [47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "n_head": [47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "embedding_s": [47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "broadcast": [47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "score_tail": [47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "n_tail": [47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "update_shard": [47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "scoring_norm": [48, 52, 53, 55, 56, 57, 58, 59], "entity_initi": [48, 49, 50, 51, 53, 55, 56, 57, 58, 59], "uniform_": 48, "relation_initi": [48, 49, 50, 51, 53, 55, 56, 57, 58, 59], "init_uniform_norm": [48, 68], "apply_tanh": 48, "dist_func_per_dim": 48, "ep": 48, "1e": 48, "06": 48, "distancebasedscorefunct": [48, 49, 50, 51, 53, 55, 56, 57, 58, 59, 68], "center": 48, "scalar": [48, 50], "bound": [48, 68], "bump": 48, "tanh": 48, "select": 48, "distanc": [48, 52, 53, 55, 56, 57, 58, 59], "whether": [48, 61], "outsid": 48, "make": [48, 61], "choic": 48, "separ": 48, "soften": 48, "geometr": 48, "width": [48, 50], "boxe_scor": 48, "bumped_ht": 48, "center_ht": 48, "width_ht": 48, "box_siz": 48, "optim": [48, 69], "emb_siz": 48, "control": 48, "broadcasted_dist": [48, 52, 53, 55, 56, 57, 58, 59], "v1": [48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 64], "v2": [48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 64], "p": [48, 52, 53, 55, 56, 57, 58, 59], "reduce_embed": [48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "v": [48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 65, 68], "init_kge_norm": [49, 68], "broadcasted_dot_product": [49, 50, 51, 54], "dot": [49, 50, 51, 54], "product": [49, 50, 51, 54], "embedding_height": 50, "embedding_width": 50, "init_xavier_norm": [50, 68], "zeros_": 50, "input_channel": 50, "output_channel": 50, "32": 50, "kernel_height": 50, "kernel_width": 50, "input_dropout": 50, "feature_map_dropout": 50, "hidden_dropout": 50, "batch_norm": 50, "layer": 50, "we": [50, 69], "By": 50, "design": [50, 68], "add_inverse_tripl": [50, 61], "create_from_dataset": [50, 61], "height": 50, "reshap": 50, "concaten": 50, "bias": 50, "channel": 50, "conv2d": 50, "kernel": 50, "rate": 50, "dropout": 50, "linear": 50, "init_kge_uniform": [51, 53, 55, 56, 57, 58, 59, 68], "normalize_ent": [53, 55, 57, 59], "l2": [53, 55, 57, 59], "auxiliari": 53, "matrix": 54, "decomposit": 54, "project": [55, 59], "init_uniform_rot": [56, 68], "real": [56, 64, 65], "tild": 57, "triplerev2": 59, "inverse_tripl": 61, "partition_mod": 61, "dummi": 61, "triple_count": 61, "triple_offset": 61, "triple_sort_idx": 61, "shard_h": 61, "shard_t": 61, "kgdataset": [61, 68], "create_from_queri": 61, "query_mod": 61, "negative_typ": 61, "n_queri": 61, "hr": 61, "rt": 61, "resp": 61, "r_inv": 61, "regular": 61, "criterion": 61, "delimit": 61, "entity_to_shard": 62, "entity_to_idx": 62, "shard_and_idx_to_ent": 62, "entity_type_count": 62, "entity_type_offset": 62, "again": 62, "n_type": 62, "npz": 62, "local_id": 62, "exclud": 62, "purpos": 63, "imaginari": [64, 65], "wise": [64, 65], "unitari": 65, "x": [66, 67], "friendli": 66, "take_along_dim": 66, "dimension": 66, "dim": 66, "take": 66, "filter_mod": 67, "compar": 67, "y": 67, "determin": 67, "look": 67, "z": 67, "spars": 67, "packag": 68, "shallow": 68, "typic": 68, "littl": 68, "perform": 68, "maxim": 68, "bandwidth": 68, "fast": 68, "leverag": 68, "achiev": 68, "introduct": 68, "overview": 68, "librari": [68, 69], "still": 68, "develop": 68, "featur": 68, "expect": 68, "overtim": 68, "occasion": 68, "bug": 68, "mai": 68, "occur": 68, "feel": 68, "report": 68, "problem": 68, "user": 68, "usag": 68, "limit": 68, "randomshardedbatchsampl": 68, "rigidshardedbatchsampl": 68, "typebasedshardednegativesampl": 68, "allscoresbess": 68, "conv": 68, "distmult": 68, "matrixdecompositionscorefunct": 68, "trans": 68, "logsigmoidloss": 68, "marginrankingloss": 68, "sampledsoftmaxcrossentropyloss": 68, "metrics_dict": 68, "initialize_entity_embed": 68, "initialize_relation_embed": 68, "refactor_embedding_shard": 68, "complex_multipl": 68, "complex_rot": 68, "gather_indic": 68, "get_entity_filt": 68, "code": 68, "server": 68, "paperspac": [68, 69], "setup": 68, "tip": 68, "bibliographi": 68, "popart": 69, "more": 69, "quick": 69, "git": 69, "import": 69, "1403": 69, "walkthrough": 69, "main": 69, "jupyt": 69, "sequenc": 69, "click": 69, "button": 69, "avail": 69, "introduc": 69, "therefor": 69, "some": 69, "approxim": 69, "estim": 69, "below": 69, "accumul": 69, "momentum": 69, "notic": 69, "cap": 69, "max": 69, "pod16": 69, "float16": 69, "sgdm": 69, "2m": 69, "2e8": 69, "13m": 69, "3e9": 69, "128": 69, "adam": 69, "4m": 69, "0e8": 69, "9m": 69, "256": 69, "ye": 69, "900k": 69, "3e8": 69, "5m": 69, "8m": 69, "2e9": 69, "512": 69, "375k": 69, "9e8": 69, "7e8": 69, "messag": 69, "about": 69, "onnx": 69, "protobuff": 69, "exceed": 69, "_popart": 69, "saveinitializerstofil": 69, "my_fil": 69}, "objects": {"": [[68, 0, 0, "-", "besskge"]], "besskge": [[5, 0, 0, "-", "batch_sampler"], [9, 0, 0, "-", "bess"], [15, 0, 0, "-", "dataset"], [17, 0, 0, "-", "embedding"], [26, 0, 0, "-", "loss"], [32, 0, 0, "-", "metric"], [38, 0, 0, "-", "negative_sampler"], [44, 0, 0, "-", "pipeline"], [46, 0, 0, "-", "scoring"], [60, 0, 0, "-", "sharding"], [63, 0, 0, "-", "utils"]], "besskge.batch_sampler": [[6, 1, 1, "", "RandomShardedBatchSampler"], [7, 1, 1, "", "RigidShardedBatchSampler"], [8, 1, 1, "", "ShardedBatchSampler"]], "besskge.batch_sampler.RandomShardedBatchSampler": [[6, 2, 1, "", "get_dataloader"], [6, 2, 1, "", "get_dataloader_sampler"], [6, 2, 1, "", "sample_triples"], [6, 2, 1, "", "worker_init_fn"]], "besskge.batch_sampler.RigidShardedBatchSampler": [[7, 2, 1, "", "get_dataloader"], [7, 2, 1, "", "get_dataloader_sampler"], [7, 2, 1, "", "sample_triples"], [7, 2, 1, "", "worker_init_fn"]], "besskge.batch_sampler.ShardedBatchSampler": [[8, 2, 1, "", "get_dataloader"], [8, 2, 1, "", "get_dataloader_sampler"], [8, 2, 1, "", "sample_triples"], [8, 2, 1, "", "worker_init_fn"]], "besskge.bess": [[10, 1, 1, "", "AllScoresBESS"], [11, 1, 1, "", "BessKGE"], [12, 1, 1, "", "EmbeddingMovingBessKGE"], [13, 1, 1, "", "ScoreMovingBessKGE"], [14, 1, 1, "", "TopKQueryBessKGE"]], "besskge.bess.AllScoresBESS": [[10, 2, 1, "", "forward"]], "besskge.bess.BessKGE": [[11, 2, 1, "", "forward"], [11, 3, 1, "", "n_embedding_parameters"], [11, 2, 1, "", "score_batch"]], "besskge.bess.EmbeddingMovingBessKGE": [[12, 2, 1, "", "forward"], [12, 3, 1, "", "n_embedding_parameters"], [12, 2, 1, "", "score_batch"]], "besskge.bess.ScoreMovingBessKGE": [[13, 2, 1, "", "forward"], [13, 3, 1, "", "n_embedding_parameters"], [13, 2, 1, "", "score_batch"]], "besskge.bess.TopKQueryBessKGE": [[14, 2, 1, "", "forward"]], "besskge.dataset": [[16, 1, 1, "", "KGDataset"]], "besskge.dataset.KGDataset": [[16, 2, 1, "", "build_ogbl_biokg"], [16, 2, 1, "", "build_ogbl_wikikg2"], [16, 2, 1, "", "build_openbiolink"], [16, 2, 1, "", "build_yago310"], [16, 4, 1, "", "entity_dict"], [16, 2, 1, "", "from_dataframe"], [16, 2, 1, "", "from_triples"], [16, 3, 1, "", "ht_types"], [16, 2, 1, "", "load"], [16, 4, 1, "", "n_entity"], [16, 4, 1, "", "n_relation_type"], [16, 4, 1, "", "neg_heads"], [16, 4, 1, "", "neg_tails"], [16, 4, 1, "", "original_triple_ids"], [16, 4, 1, "", "relation_dict"], [16, 2, 1, "", "save"], [16, 4, 1, "", "triples"], [16, 4, 1, "", "type_offsets"]], "besskge.embedding": [[18, 5, 1, "", "init_KGE_normal"], [19, 5, 1, "", "init_KGE_uniform"], [20, 5, 1, "", "init_uniform_norm"], [21, 5, 1, "", "init_uniform_rotation"], [22, 5, 1, "", "init_xavier_norm"], [23, 5, 1, "", "initialize_entity_embedding"], [24, 5, 1, "", "initialize_relation_embedding"], [25, 5, 1, "", "refactor_embedding_sharding"]], "besskge.loss": [[27, 1, 1, "", "BaseLossFunction"], [28, 1, 1, "", "LogSigmoidLoss"], [29, 1, 1, "", "MarginBasedLossFunction"], [30, 1, 1, "", "MarginRankingLoss"], [31, 1, 1, "", "SampledSoftmaxCrossEntropyLoss"]], "besskge.loss.BaseLossFunction": [[27, 2, 1, "", "forward"], [27, 2, 1, "", "get_negative_weights"], [27, 4, 1, "", "loss_scale"], [27, 4, 1, "", "negative_adversarial_sampling"], [27, 4, 1, "", "negative_adversarial_scale"]], "besskge.loss.LogSigmoidLoss": [[28, 2, 1, "", "forward"], [28, 2, 1, "", "get_negative_weights"], [28, 4, 1, "", "loss_scale"], [28, 4, 1, "", "negative_adversarial_sampling"], [28, 4, 1, "", "negative_adversarial_scale"]], "besskge.loss.MarginBasedLossFunction": [[29, 2, 1, "", "forward"], [29, 2, 1, "", "get_negative_weights"], [29, 4, 1, "", "loss_scale"], [29, 4, 1, "", "negative_adversarial_sampling"], [29, 4, 1, "", "negative_adversarial_scale"]], "besskge.loss.MarginRankingLoss": [[30, 2, 1, "", "forward"], [30, 2, 1, "", "get_negative_weights"], [30, 4, 1, "", "loss_scale"], [30, 4, 1, "", "negative_adversarial_sampling"], [30, 4, 1, "", "negative_adversarial_scale"]], "besskge.loss.SampledSoftmaxCrossEntropyLoss": [[31, 2, 1, "", "forward"], [31, 2, 1, "", "get_negative_weights"], [31, 4, 1, "", "loss_scale"], [31, 4, 1, "", "negative_adversarial_sampling"], [31, 4, 1, "", "negative_adversarial_scale"]], "besskge.metric": [[33, 1, 1, "", "BaseMetric"], [34, 1, 1, "", "Evaluation"], [35, 1, 1, "", "HitsAtK"], [36, 6, 1, "", "METRICS_DICT"], [37, 1, 1, "", "ReciprocalRank"]], "besskge.metric.Evaluation": [[34, 2, 1, "", "dict_metrics_from_ranks"], [34, 2, 1, "", "ranks_from_indices"], [34, 2, 1, "", "ranks_from_scores"], [34, 2, 1, "", "stacked_metrics_from_ranks"]], "besskge.negative_sampler": [[39, 1, 1, "", "PlaceholderNegativeSampler"], [40, 1, 1, "", "RandomShardedNegativeSampler"], [41, 1, 1, "", "ShardedNegativeSampler"], [42, 1, 1, "", "TripleBasedShardedNegativeSampler"], [43, 1, 1, "", "TypeBasedShardedNegativeSampler"]], "besskge.negative_sampler.PlaceholderNegativeSampler": [[39, 4, 1, "", "corruption_scheme"], [39, 4, 1, "", "flat_negative_format"], [39, 4, 1, "", "local_sampling"], [39, 4, 1, "", "rng"]], "besskge.negative_sampler.RandomShardedNegativeSampler": [[40, 4, 1, "", "corruption_scheme"], [40, 4, 1, "", "flat_negative_format"], [40, 4, 1, "", "local_sampling"], [40, 4, 1, "", "rng"]], "besskge.negative_sampler.ShardedNegativeSampler": [[41, 4, 1, "", "corruption_scheme"], [41, 4, 1, "", "flat_negative_format"], [41, 4, 1, "", "local_sampling"], [41, 4, 1, "", "rng"]], "besskge.negative_sampler.TripleBasedShardedNegativeSampler": [[42, 4, 1, "", "corruption_scheme"], [42, 4, 1, "", "flat_negative_format"], [42, 4, 1, "", "local_sampling"], [42, 2, 1, "", "pad_negatives"], [42, 4, 1, "", "rng"], [42, 2, 1, "", "shard_negatives"]], "besskge.negative_sampler.TypeBasedShardedNegativeSampler": [[43, 4, 1, "", "corruption_scheme"], [43, 4, 1, "", "flat_negative_format"], [43, 4, 1, "", "local_sampling"], [43, 4, 1, "", "rng"]], "besskge.pipeline": [[45, 1, 1, "", "AllScoresPipeline"]], "besskge.pipeline.AllScoresPipeline": [[45, 2, 1, "", "forward"]], "besskge.scoring": [[47, 1, 1, "", "BaseScoreFunction"], [48, 1, 1, "", "BoxE"], [49, 1, 1, "", "ComplEx"], [50, 1, 1, "", "ConvE"], [51, 1, 1, "", "DistMult"], [52, 1, 1, "", "DistanceBasedScoreFunction"], [53, 1, 1, "", "InterHT"], [54, 1, 1, "", "MatrixDecompositionScoreFunction"], [55, 1, 1, "", "PairRE"], [56, 1, 1, "", "RotatE"], [57, 1, 1, "", "TranS"], [58, 1, 1, "", "TransE"], [59, 1, 1, "", "TripleRE"]], "besskge.scoring.BaseScoreFunction": [[47, 4, 1, "", "entity_embedding"], [47, 2, 1, "", "forward"], [47, 4, 1, "", "negative_sample_sharing"], [47, 4, 1, "", "relation_embedding"], [47, 2, 1, "", "score_heads"], [47, 2, 1, "", "score_tails"], [47, 2, 1, "", "score_triple"], [47, 4, 1, "", "sharding"], [47, 2, 1, "", "update_sharding"]], "besskge.scoring.BoxE": [[48, 2, 1, "", "boxe_score"], [48, 2, 1, "", "broadcasted_distance"], [48, 4, 1, "", "entity_embedding"], [48, 2, 1, "", "forward"], [48, 4, 1, "", "negative_sample_sharing"], [48, 2, 1, "", "reduce_embedding"], [48, 4, 1, "", "relation_embedding"], [48, 2, 1, "", "score_heads"], [48, 2, 1, "", "score_tails"], [48, 2, 1, "", "score_triple"], [48, 4, 1, "", "sharding"], [48, 2, 1, "", "update_sharding"]], "besskge.scoring.ComplEx": [[49, 2, 1, "", "broadcasted_dot_product"], [49, 4, 1, "", "entity_embedding"], [49, 2, 1, "", "forward"], [49, 4, 1, "", "negative_sample_sharing"], [49, 2, 1, "", "reduce_embedding"], [49, 4, 1, "", "relation_embedding"], [49, 2, 1, "", "score_heads"], [49, 2, 1, "", "score_tails"], [49, 2, 1, "", "score_triple"], [49, 4, 1, "", "sharding"], [49, 2, 1, "", "update_sharding"]], "besskge.scoring.ConvE": [[50, 2, 1, "", "broadcasted_dot_product"], [50, 4, 1, "", "entity_embedding"], [50, 2, 1, "", "forward"], [50, 4, 1, "", "negative_sample_sharing"], [50, 2, 1, "", "reduce_embedding"], [50, 4, 1, "", "relation_embedding"], [50, 2, 1, "", "score_heads"], [50, 2, 1, "", "score_tails"], [50, 2, 1, "", "score_triple"], [50, 4, 1, "", "sharding"], [50, 2, 1, "", "update_sharding"]], "besskge.scoring.DistMult": [[51, 2, 1, "", "broadcasted_dot_product"], [51, 4, 1, "", "entity_embedding"], [51, 2, 1, "", "forward"], [51, 4, 1, "", "negative_sample_sharing"], [51, 2, 1, "", "reduce_embedding"], [51, 4, 1, "", "relation_embedding"], [51, 2, 1, "", "score_heads"], [51, 2, 1, "", "score_tails"], [51, 2, 1, "", "score_triple"], [51, 4, 1, "", "sharding"], [51, 2, 1, "", "update_sharding"]], "besskge.scoring.DistanceBasedScoreFunction": [[52, 2, 1, "", "broadcasted_distance"], [52, 4, 1, "", "entity_embedding"], [52, 2, 1, "", "forward"], [52, 4, 1, "", "negative_sample_sharing"], [52, 2, 1, "", "reduce_embedding"], [52, 4, 1, "", "relation_embedding"], [52, 2, 1, "", "score_heads"], [52, 2, 1, "", "score_tails"], [52, 2, 1, "", "score_triple"], [52, 4, 1, "", "sharding"], [52, 2, 1, "", "update_sharding"]], "besskge.scoring.InterHT": [[53, 2, 1, "", "broadcasted_distance"], [53, 4, 1, "", "entity_embedding"], [53, 2, 1, "", "forward"], [53, 4, 1, "", "negative_sample_sharing"], [53, 2, 1, "", "reduce_embedding"], [53, 4, 1, "", "relation_embedding"], [53, 2, 1, "", "score_heads"], [53, 2, 1, "", "score_tails"], [53, 2, 1, "", "score_triple"], [53, 4, 1, "", "sharding"], [53, 2, 1, "", "update_sharding"]], "besskge.scoring.MatrixDecompositionScoreFunction": [[54, 2, 1, "", "broadcasted_dot_product"], [54, 4, 1, "", "entity_embedding"], [54, 2, 1, "", "forward"], [54, 4, 1, "", "negative_sample_sharing"], [54, 2, 1, "", "reduce_embedding"], [54, 4, 1, "", "relation_embedding"], [54, 2, 1, "", "score_heads"], [54, 2, 1, "", "score_tails"], [54, 2, 1, "", "score_triple"], [54, 4, 1, "", "sharding"], [54, 2, 1, "", "update_sharding"]], "besskge.scoring.PairRE": [[55, 2, 1, "", "broadcasted_distance"], [55, 4, 1, "", "entity_embedding"], [55, 2, 1, "", "forward"], [55, 4, 1, "", "negative_sample_sharing"], [55, 2, 1, "", "reduce_embedding"], [55, 4, 1, "", "relation_embedding"], [55, 2, 1, "", "score_heads"], [55, 2, 1, "", "score_tails"], [55, 2, 1, "", "score_triple"], [55, 4, 1, "", "sharding"], [55, 2, 1, "", "update_sharding"]], "besskge.scoring.RotatE": [[56, 2, 1, "", "broadcasted_distance"], [56, 4, 1, "", "entity_embedding"], [56, 2, 1, "", "forward"], [56, 4, 1, "", "negative_sample_sharing"], [56, 2, 1, "", "reduce_embedding"], [56, 4, 1, "", "relation_embedding"], [56, 2, 1, "", "score_heads"], [56, 2, 1, "", "score_tails"], [56, 2, 1, "", "score_triple"], [56, 4, 1, "", "sharding"], [56, 2, 1, "", "update_sharding"]], "besskge.scoring.TranS": [[57, 2, 1, "", "broadcasted_distance"], [57, 4, 1, "", "entity_embedding"], [57, 2, 1, "", "forward"], [57, 4, 1, "", "negative_sample_sharing"], [57, 2, 1, "", "reduce_embedding"], [57, 4, 1, "", "relation_embedding"], [57, 2, 1, "", "score_heads"], [57, 2, 1, "", "score_tails"], [57, 2, 1, "", "score_triple"], [57, 4, 1, "", "sharding"], [57, 2, 1, "", "update_sharding"]], "besskge.scoring.TransE": [[58, 2, 1, "", "broadcasted_distance"], [58, 4, 1, "", "entity_embedding"], [58, 2, 1, "", "forward"], [58, 4, 1, "", "negative_sample_sharing"], [58, 2, 1, "", "reduce_embedding"], [58, 4, 1, "", "relation_embedding"], [58, 2, 1, "", "score_heads"], [58, 2, 1, "", "score_tails"], [58, 2, 1, "", "score_triple"], [58, 4, 1, "", "sharding"], [58, 2, 1, "", "update_sharding"]], "besskge.scoring.TripleRE": [[59, 2, 1, "", "broadcasted_distance"], [59, 4, 1, "", "entity_embedding"], [59, 2, 1, "", "forward"], [59, 4, 1, "", "negative_sample_sharing"], [59, 2, 1, "", "reduce_embedding"], [59, 4, 1, "", "relation_embedding"], [59, 2, 1, "", "score_heads"], [59, 2, 1, "", "score_tails"], [59, 2, 1, "", "score_triple"], [59, 4, 1, "", "sharding"], [59, 2, 1, "", "update_sharding"]], "besskge.sharding": [[61, 1, 1, "", "PartitionedTripleSet"], [62, 1, 1, "", "Sharding"]], "besskge.sharding.PartitionedTripleSet": [[61, 2, 1, "", "create_from_dataset"], [61, 2, 1, "", "create_from_queries"], [61, 4, 1, "", "dummy"], [61, 4, 1, "", "inverse_triples"], [61, 4, 1, "", "neg_heads"], [61, 4, 1, "", "neg_tails"], [61, 4, 1, "", "partition_mode"], [61, 4, 1, "", "sharding"], [61, 4, 1, "", "triple_counts"], [61, 4, 1, "", "triple_offsets"], [61, 4, 1, "", "triple_sort_idx"], [61, 4, 1, "", "triples"], [61, 4, 1, "", "types"]], "besskge.sharding.Sharding": [[62, 2, 1, "", "create"], [62, 4, 1, "", "entity_to_idx"], [62, 4, 1, "", "entity_to_shard"], [62, 4, 1, "", "entity_type_counts"], [62, 4, 1, "", "entity_type_offsets"], [62, 2, 1, "", "load"], [62, 3, 1, "", "max_entity_per_shard"], [62, 3, 1, "", "n_entity"], [62, 4, 1, "", "n_shard"], [62, 2, 1, "", "save"], [62, 4, 1, "", "shard_and_idx_to_entity"], [62, 4, 1, "", "shard_counts"]], "besskge.utils": [[64, 5, 1, "", "complex_multiplication"], [65, 5, 1, "", "complex_rotation"], [66, 5, 1, "", "gather_indices"], [67, 5, 1, "", "get_entity_filter"]]}, "objtypes": {"0": "py:module", "1": "py:class", "2": "py:method", "3": "py:property", "4": "py:attribute", "5": "py:function", "6": "py:data"}, "objnames": {"0": ["py", "module", "Python module"], "1": ["py", "class", "Python class"], "2": ["py", "method", "Python method"], "3": ["py", "property", "Python property"], "4": ["py", "attribute", "Python attribute"], "5": ["py", "function", "Python function"], "6": ["py", "data", "Python data"]}, "titleterms": {"bess": [0, 1, 3, 4, 9, 10, 11, 12, 13, 14, 68], "kge": [0, 3, 4, 68], "api": 0, "refer": 0, "overview": 1, "bibliographi": 2, "how": [3, 4], "contribut": [3, 4], "project": [3, 4], "v": [3, 4], "code": [3, 4], "server": [3, 4], "paperspac": [3, 4], "setup": [3, 4], "local": [3, 4], "machin": [3, 4], "develop": [3, 4], "tip": [3, 4], "besskg": [5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67], "batch_sampl": [5, 6, 7, 8], "randomshardedbatchsampl": 6, "rigidshardedbatchsampl": 7, "shardedbatchsampl": 8, "allscoresbess": 10, "embeddingmovingbesskg": 12, "scoremovingbesskg": 13, "topkquerybesskg": 14, "dataset": [15, 16], "kgdataset": 16, "embed": [17, 18, 19, 20, 21, 22, 23, 24, 25], "init_kge_norm": 18, "init_kge_uniform": 19, "init_uniform_norm": 20, "init_uniform_rot": 21, "init_xavier_norm": 22, "initialize_entity_embed": 23, "initialize_relation_embed": 24, "refactor_embedding_shard": 25, "loss": [26, 27, 28, 29, 30, 31], "baselossfunct": 27, "logsigmoidloss": 28, "marginbasedlossfunct": 29, "marginrankingloss": 30, "sampledsoftmaxcrossentropyloss": 31, "metric": [32, 33, 34, 35, 36, 37], "basemetr": 33, "evalu": 34, "hitsatk": 35, "metrics_dict": 36, "reciprocalrank": 37, "negative_sampl": [38, 39, 40, 41, 42, 43], "placeholdernegativesampl": 39, "randomshardednegativesampl": 40, "shardednegativesampl": 41, "triplebasedshardednegativesampl": 42, "typebasedshardednegativesampl": 43, "pipelin": [44, 45], "allscorespipelin": 45, "score": [46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59], "basescorefunct": 47, "box": 48, "complex": 49, "conv": 50, "distmult": 51, "distancebasedscorefunct": 52, "interht": 53, "matrixdecompositionscorefunct": 54, "pairr": 55, "rotat": 56, "tran": 57, "trans": 58, "tripler": 59, "shard": [60, 61, 62], "partitionedtripleset": 61, "util": [63, 64, 65, 66, 67], "complex_multipl": 64, "complex_rot": 65, "gather_indic": 66, "get_entity_filt": 67, "content": 68, "user": 69, "guid": 69, "instal": 69, "usag": 69, "get": 69, "start": 69, "limit": 69}, "envversion": {"sphinx.domains.c": 2, "sphinx.domains.changeset": 1, "sphinx.domains.citation": 1, "sphinx.domains.cpp": 8, "sphinx.domains.index": 1, "sphinx.domains.javascript": 2, "sphinx.domains.math": 2, "sphinx.domains.python": 3, "sphinx.domains.rst": 2, "sphinx.domains.std": 2, "sphinx.ext.todo": 2, "sphinx.ext.viewcode": 1, "sphinx.ext.intersphinx": 1, "sphinxcontrib.bibtex": 9, "sphinx": 57}, "alltitles": {"BESS-KGE API Reference": [[0, "bess-kge-api-reference"]], "BESS overview": [[1, "bess-overview"]], "Bibliography": [[2, "bibliography"]], "How to contribute to the BESS-KGE project": [[3, "how-to-contribute-to-the-bess-kge-project"], [4, "how-to-contribute-to-the-bess-kge-project"]], "VS Code server on Paperspace": [[3, "vs-code-server-on-paperspace"], [4, "vs-code-server-on-paperspace"]], "Setup on local machine": [[3, "setup-on-local-machine"], [4, "setup-on-local-machine"]], "Development tips": [[3, "development-tips"], [4, "development-tips"]], "besskge.batch_sampler": [[5, "module-besskge.batch_sampler"]], "besskge.batch_sampler.RandomShardedBatchSampler": [[6, "besskge-batch-sampler-randomshardedbatchsampler"]], "besskge.batch_sampler.RigidShardedBatchSampler": [[7, "besskge-batch-sampler-rigidshardedbatchsampler"]], "besskge.batch_sampler.ShardedBatchSampler": [[8, "besskge-batch-sampler-shardedbatchsampler"]], "besskge.bess": [[9, "module-besskge.bess"]], "besskge.bess.AllScoresBESS": [[10, "besskge-bess-allscoresbess"]], "besskge.bess.BessKGE": [[11, "besskge-bess-besskge"]], "besskge.bess.EmbeddingMovingBessKGE": [[12, "besskge-bess-embeddingmovingbesskge"]], "besskge.bess.ScoreMovingBessKGE": [[13, "besskge-bess-scoremovingbesskge"]], "besskge.bess.TopKQueryBessKGE": [[14, "besskge-bess-topkquerybesskge"]], "besskge.dataset": [[15, "module-besskge.dataset"]], "besskge.dataset.KGDataset": [[16, "besskge-dataset-kgdataset"]], "besskge.embedding": [[17, "module-besskge.embedding"]], "besskge.embedding.init_KGE_normal": [[18, "besskge-embedding-init-kge-normal"]], "besskge.embedding.init_KGE_uniform": [[19, "besskge-embedding-init-kge-uniform"]], "besskge.embedding.init_uniform_norm": [[20, "besskge-embedding-init-uniform-norm"]], "besskge.embedding.init_uniform_rotation": [[21, "besskge-embedding-init-uniform-rotation"]], "besskge.embedding.init_xavier_norm": [[22, "besskge-embedding-init-xavier-norm"]], "besskge.embedding.initialize_entity_embedding": [[23, "besskge-embedding-initialize-entity-embedding"]], "besskge.embedding.initialize_relation_embedding": [[24, "besskge-embedding-initialize-relation-embedding"]], "besskge.embedding.refactor_embedding_sharding": [[25, "besskge-embedding-refactor-embedding-sharding"]], "besskge.loss": [[26, "module-besskge.loss"]], "besskge.loss.BaseLossFunction": [[27, "besskge-loss-baselossfunction"]], "besskge.loss.LogSigmoidLoss": [[28, "besskge-loss-logsigmoidloss"]], "besskge.loss.MarginBasedLossFunction": [[29, "besskge-loss-marginbasedlossfunction"]], "besskge.loss.MarginRankingLoss": [[30, "besskge-loss-marginrankingloss"]], "besskge.loss.SampledSoftmaxCrossEntropyLoss": [[31, "besskge-loss-sampledsoftmaxcrossentropyloss"]], "besskge.metric": [[32, "module-besskge.metric"]], "besskge.metric.BaseMetric": [[33, "besskge-metric-basemetric"]], "besskge.metric.Evaluation": [[34, "besskge-metric-evaluation"]], "besskge.metric.HitsAtK": [[35, "besskge-metric-hitsatk"]], "besskge.metric.METRICS_DICT": [[36, "besskge-metric-metrics-dict"]], "besskge.metric.ReciprocalRank": [[37, "besskge-metric-reciprocalrank"]], "besskge.negative_sampler": [[38, "module-besskge.negative_sampler"]], "besskge.negative_sampler.PlaceholderNegativeSampler": [[39, "besskge-negative-sampler-placeholdernegativesampler"]], "besskge.negative_sampler.RandomShardedNegativeSampler": [[40, "besskge-negative-sampler-randomshardednegativesampler"]], "besskge.negative_sampler.ShardedNegativeSampler": [[41, "besskge-negative-sampler-shardednegativesampler"]], "besskge.negative_sampler.TripleBasedShardedNegativeSampler": [[42, "besskge-negative-sampler-triplebasedshardednegativesampler"]], "besskge.negative_sampler.TypeBasedShardedNegativeSampler": [[43, "besskge-negative-sampler-typebasedshardednegativesampler"]], "besskge.pipeline": [[44, "module-besskge.pipeline"]], "besskge.pipeline.AllScoresPipeline": [[45, "besskge-pipeline-allscorespipeline"]], "besskge.scoring": [[46, "module-besskge.scoring"]], "besskge.scoring.BaseScoreFunction": [[47, "besskge-scoring-basescorefunction"]], "besskge.scoring.BoxE": [[48, "besskge-scoring-boxe"]], "besskge.scoring.ComplEx": [[49, "besskge-scoring-complex"]], "besskge.scoring.ConvE": [[50, "besskge-scoring-conve"]], "besskge.scoring.DistMult": [[51, "besskge-scoring-distmult"]], "besskge.scoring.DistanceBasedScoreFunction": [[52, "besskge-scoring-distancebasedscorefunction"]], "besskge.scoring.InterHT": [[53, "besskge-scoring-interht"]], "besskge.scoring.MatrixDecompositionScoreFunction": [[54, "besskge-scoring-matrixdecompositionscorefunction"]], "besskge.scoring.PairRE": [[55, "besskge-scoring-pairre"]], "besskge.scoring.RotatE": [[56, "besskge-scoring-rotate"]], "besskge.scoring.TranS": [[57, "besskge-scoring-trans"]], "besskge.scoring.TransE": [[58, "besskge-scoring-transe"]], "besskge.scoring.TripleRE": [[59, "besskge-scoring-triplere"]], "besskge.sharding": [[60, "module-besskge.sharding"]], "besskge.sharding.PartitionedTripleSet": [[61, "besskge-sharding-partitionedtripleset"]], "besskge.sharding.Sharding": [[62, "besskge-sharding-sharding"]], "besskge.utils": [[63, "module-besskge.utils"]], "besskge.utils.complex_multiplication": [[64, "besskge-utils-complex-multiplication"]], "besskge.utils.complex_rotation": [[65, "besskge-utils-complex-rotation"]], "besskge.utils.gather_indices": [[66, "besskge-utils-gather-indices"]], "besskge.utils.get_entity_filter": [[67, "besskge-utils-get-entity-filter"]], "BESS-KGE": [[68, "module-besskge"]], "Contents": [[68, null]], "User guide": [[69, "user-guide"]], "Installation and usage": [[69, "installation-and-usage"]], "Getting started": [[69, "getting-started"]], "Limitations": [[69, "limitations"]]}, "indexentries": {"besskge.batch_sampler": [[5, "module-besskge.batch_sampler"]], "module": [[5, "module-besskge.batch_sampler"], [9, "module-besskge.bess"], [15, "module-besskge.dataset"], [17, "module-besskge.embedding"], [26, "module-besskge.loss"], [32, "module-besskge.metric"], [38, "module-besskge.negative_sampler"], [44, "module-besskge.pipeline"], [46, "module-besskge.scoring"], [60, "module-besskge.sharding"], [63, "module-besskge.utils"], [68, "module-besskge"]], "randomshardedbatchsampler (class in besskge.batch_sampler)": [[6, "besskge.batch_sampler.RandomShardedBatchSampler"]], "get_dataloader() (besskge.batch_sampler.randomshardedbatchsampler method)": [[6, "besskge.batch_sampler.RandomShardedBatchSampler.get_dataloader"]], "get_dataloader_sampler() (besskge.batch_sampler.randomshardedbatchsampler method)": [[6, "besskge.batch_sampler.RandomShardedBatchSampler.get_dataloader_sampler"]], "sample_triples() (besskge.batch_sampler.randomshardedbatchsampler method)": [[6, "besskge.batch_sampler.RandomShardedBatchSampler.sample_triples"]], "worker_init_fn() (besskge.batch_sampler.randomshardedbatchsampler static method)": [[6, "besskge.batch_sampler.RandomShardedBatchSampler.worker_init_fn"]], "rigidshardedbatchsampler (class in besskge.batch_sampler)": [[7, "besskge.batch_sampler.RigidShardedBatchSampler"]], "get_dataloader() (besskge.batch_sampler.rigidshardedbatchsampler method)": [[7, "besskge.batch_sampler.RigidShardedBatchSampler.get_dataloader"]], "get_dataloader_sampler() (besskge.batch_sampler.rigidshardedbatchsampler method)": [[7, "besskge.batch_sampler.RigidShardedBatchSampler.get_dataloader_sampler"]], "sample_triples() (besskge.batch_sampler.rigidshardedbatchsampler method)": [[7, "besskge.batch_sampler.RigidShardedBatchSampler.sample_triples"]], "worker_init_fn() (besskge.batch_sampler.rigidshardedbatchsampler static method)": [[7, "besskge.batch_sampler.RigidShardedBatchSampler.worker_init_fn"]], "shardedbatchsampler (class in besskge.batch_sampler)": [[8, "besskge.batch_sampler.ShardedBatchSampler"]], "get_dataloader() (besskge.batch_sampler.shardedbatchsampler method)": [[8, "besskge.batch_sampler.ShardedBatchSampler.get_dataloader"]], "get_dataloader_sampler() (besskge.batch_sampler.shardedbatchsampler method)": [[8, "besskge.batch_sampler.ShardedBatchSampler.get_dataloader_sampler"]], "sample_triples() (besskge.batch_sampler.shardedbatchsampler method)": [[8, "besskge.batch_sampler.ShardedBatchSampler.sample_triples"]], "worker_init_fn() (besskge.batch_sampler.shardedbatchsampler static method)": [[8, "besskge.batch_sampler.ShardedBatchSampler.worker_init_fn"]], "besskge.bess": [[9, "module-besskge.bess"]], "allscoresbess (class in besskge.bess)": [[10, "besskge.bess.AllScoresBESS"]], "forward() (besskge.bess.allscoresbess method)": [[10, "besskge.bess.AllScoresBESS.forward"]], "besskge (class in besskge.bess)": [[11, "besskge.bess.BessKGE"]], "forward() (besskge.bess.besskge method)": [[11, "besskge.bess.BessKGE.forward"]], "n_embedding_parameters (besskge.bess.besskge property)": [[11, "besskge.bess.BessKGE.n_embedding_parameters"]], "score_batch() (besskge.bess.besskge method)": [[11, "besskge.bess.BessKGE.score_batch"]], "embeddingmovingbesskge (class in besskge.bess)": [[12, "besskge.bess.EmbeddingMovingBessKGE"]], "forward() (besskge.bess.embeddingmovingbesskge method)": [[12, "besskge.bess.EmbeddingMovingBessKGE.forward"]], "n_embedding_parameters (besskge.bess.embeddingmovingbesskge property)": [[12, "besskge.bess.EmbeddingMovingBessKGE.n_embedding_parameters"]], "score_batch() (besskge.bess.embeddingmovingbesskge method)": [[12, "besskge.bess.EmbeddingMovingBessKGE.score_batch"]], "scoremovingbesskge (class in besskge.bess)": [[13, "besskge.bess.ScoreMovingBessKGE"]], "forward() (besskge.bess.scoremovingbesskge method)": [[13, "besskge.bess.ScoreMovingBessKGE.forward"]], "n_embedding_parameters (besskge.bess.scoremovingbesskge property)": [[13, "besskge.bess.ScoreMovingBessKGE.n_embedding_parameters"]], "score_batch() (besskge.bess.scoremovingbesskge method)": [[13, "besskge.bess.ScoreMovingBessKGE.score_batch"]], "topkquerybesskge (class in besskge.bess)": [[14, "besskge.bess.TopKQueryBessKGE"]], "forward() (besskge.bess.topkquerybesskge method)": [[14, "besskge.bess.TopKQueryBessKGE.forward"]], "besskge.dataset": [[15, "module-besskge.dataset"]], "kgdataset (class in besskge.dataset)": [[16, "besskge.dataset.KGDataset"]], "build_ogbl_biokg() (besskge.dataset.kgdataset class method)": [[16, "besskge.dataset.KGDataset.build_ogbl_biokg"]], "build_ogbl_wikikg2() (besskge.dataset.kgdataset class method)": [[16, "besskge.dataset.KGDataset.build_ogbl_wikikg2"]], "build_openbiolink() (besskge.dataset.kgdataset class method)": [[16, "besskge.dataset.KGDataset.build_openbiolink"]], "build_yago310() (besskge.dataset.kgdataset class method)": [[16, "besskge.dataset.KGDataset.build_yago310"]], "entity_dict (besskge.dataset.kgdataset attribute)": [[16, "besskge.dataset.KGDataset.entity_dict"]], "from_dataframe() (besskge.dataset.kgdataset class method)": [[16, "besskge.dataset.KGDataset.from_dataframe"]], "from_triples() (besskge.dataset.kgdataset class method)": [[16, "besskge.dataset.KGDataset.from_triples"]], "ht_types (besskge.dataset.kgdataset property)": [[16, "besskge.dataset.KGDataset.ht_types"]], "load() (besskge.dataset.kgdataset class method)": [[16, "besskge.dataset.KGDataset.load"]], "n_entity (besskge.dataset.kgdataset attribute)": [[16, "besskge.dataset.KGDataset.n_entity"]], "n_relation_type (besskge.dataset.kgdataset attribute)": [[16, "besskge.dataset.KGDataset.n_relation_type"]], "neg_heads (besskge.dataset.kgdataset attribute)": [[16, "besskge.dataset.KGDataset.neg_heads"]], "neg_tails (besskge.dataset.kgdataset attribute)": [[16, "besskge.dataset.KGDataset.neg_tails"]], "original_triple_ids (besskge.dataset.kgdataset attribute)": [[16, "besskge.dataset.KGDataset.original_triple_ids"]], "relation_dict (besskge.dataset.kgdataset attribute)": [[16, "besskge.dataset.KGDataset.relation_dict"]], "save() (besskge.dataset.kgdataset method)": [[16, "besskge.dataset.KGDataset.save"]], "triples (besskge.dataset.kgdataset attribute)": [[16, "besskge.dataset.KGDataset.triples"]], "type_offsets (besskge.dataset.kgdataset attribute)": [[16, "besskge.dataset.KGDataset.type_offsets"]], "besskge.embedding": [[17, "module-besskge.embedding"]], "init_kge_normal() (in module besskge.embedding)": [[18, "besskge.embedding.init_KGE_normal"]], "init_kge_uniform() (in module besskge.embedding)": [[19, "besskge.embedding.init_KGE_uniform"]], "init_uniform_norm() (in module besskge.embedding)": [[20, "besskge.embedding.init_uniform_norm"]], "init_uniform_rotation() (in module besskge.embedding)": [[21, "besskge.embedding.init_uniform_rotation"]], "init_xavier_norm() (in module besskge.embedding)": [[22, "besskge.embedding.init_xavier_norm"]], "initialize_entity_embedding() (in module besskge.embedding)": [[23, "besskge.embedding.initialize_entity_embedding"]], "initialize_relation_embedding() (in module besskge.embedding)": [[24, "besskge.embedding.initialize_relation_embedding"]], "refactor_embedding_sharding() (in module besskge.embedding)": [[25, "besskge.embedding.refactor_embedding_sharding"]], "besskge.loss": [[26, "module-besskge.loss"]], "baselossfunction (class in besskge.loss)": [[27, "besskge.loss.BaseLossFunction"]], "forward() (besskge.loss.baselossfunction method)": [[27, "besskge.loss.BaseLossFunction.forward"]], "get_negative_weights() (besskge.loss.baselossfunction method)": [[27, "besskge.loss.BaseLossFunction.get_negative_weights"]], "loss_scale (besskge.loss.baselossfunction attribute)": [[27, "besskge.loss.BaseLossFunction.loss_scale"]], "negative_adversarial_sampling (besskge.loss.baselossfunction attribute)": [[27, "besskge.loss.BaseLossFunction.negative_adversarial_sampling"]], "negative_adversarial_scale (besskge.loss.baselossfunction attribute)": [[27, "besskge.loss.BaseLossFunction.negative_adversarial_scale"]], "logsigmoidloss (class in besskge.loss)": [[28, "besskge.loss.LogSigmoidLoss"]], "forward() (besskge.loss.logsigmoidloss method)": [[28, "besskge.loss.LogSigmoidLoss.forward"]], "get_negative_weights() (besskge.loss.logsigmoidloss method)": [[28, "besskge.loss.LogSigmoidLoss.get_negative_weights"]], "loss_scale (besskge.loss.logsigmoidloss attribute)": [[28, "besskge.loss.LogSigmoidLoss.loss_scale"]], "negative_adversarial_sampling (besskge.loss.logsigmoidloss attribute)": [[28, "besskge.loss.LogSigmoidLoss.negative_adversarial_sampling"]], "negative_adversarial_scale (besskge.loss.logsigmoidloss attribute)": [[28, "besskge.loss.LogSigmoidLoss.negative_adversarial_scale"]], "marginbasedlossfunction (class in besskge.loss)": [[29, "besskge.loss.MarginBasedLossFunction"]], "forward() (besskge.loss.marginbasedlossfunction method)": [[29, "besskge.loss.MarginBasedLossFunction.forward"]], "get_negative_weights() (besskge.loss.marginbasedlossfunction method)": [[29, "besskge.loss.MarginBasedLossFunction.get_negative_weights"]], "loss_scale (besskge.loss.marginbasedlossfunction attribute)": [[29, "besskge.loss.MarginBasedLossFunction.loss_scale"]], "negative_adversarial_sampling (besskge.loss.marginbasedlossfunction attribute)": [[29, "besskge.loss.MarginBasedLossFunction.negative_adversarial_sampling"]], "negative_adversarial_scale (besskge.loss.marginbasedlossfunction attribute)": [[29, "besskge.loss.MarginBasedLossFunction.negative_adversarial_scale"]], "marginrankingloss (class in besskge.loss)": [[30, "besskge.loss.MarginRankingLoss"]], "forward() (besskge.loss.marginrankingloss method)": [[30, "besskge.loss.MarginRankingLoss.forward"]], "get_negative_weights() (besskge.loss.marginrankingloss method)": [[30, "besskge.loss.MarginRankingLoss.get_negative_weights"]], "loss_scale (besskge.loss.marginrankingloss attribute)": [[30, "besskge.loss.MarginRankingLoss.loss_scale"]], "negative_adversarial_sampling (besskge.loss.marginrankingloss attribute)": [[30, "besskge.loss.MarginRankingLoss.negative_adversarial_sampling"]], "negative_adversarial_scale (besskge.loss.marginrankingloss attribute)": [[30, "besskge.loss.MarginRankingLoss.negative_adversarial_scale"]], "sampledsoftmaxcrossentropyloss (class in besskge.loss)": [[31, "besskge.loss.SampledSoftmaxCrossEntropyLoss"]], "forward() (besskge.loss.sampledsoftmaxcrossentropyloss method)": [[31, "besskge.loss.SampledSoftmaxCrossEntropyLoss.forward"]], "get_negative_weights() (besskge.loss.sampledsoftmaxcrossentropyloss method)": [[31, "besskge.loss.SampledSoftmaxCrossEntropyLoss.get_negative_weights"]], "loss_scale (besskge.loss.sampledsoftmaxcrossentropyloss attribute)": [[31, "besskge.loss.SampledSoftmaxCrossEntropyLoss.loss_scale"]], "negative_adversarial_sampling (besskge.loss.sampledsoftmaxcrossentropyloss attribute)": [[31, "besskge.loss.SampledSoftmaxCrossEntropyLoss.negative_adversarial_sampling"]], "negative_adversarial_scale (besskge.loss.sampledsoftmaxcrossentropyloss attribute)": [[31, "besskge.loss.SampledSoftmaxCrossEntropyLoss.negative_adversarial_scale"]], "besskge.metric": [[32, "module-besskge.metric"]], "basemetric (class in besskge.metric)": [[33, "besskge.metric.BaseMetric"]], "evaluation (class in besskge.metric)": [[34, "besskge.metric.Evaluation"]], "dict_metrics_from_ranks() (besskge.metric.evaluation method)": [[34, "besskge.metric.Evaluation.dict_metrics_from_ranks"]], "ranks_from_indices() (besskge.metric.evaluation method)": [[34, "besskge.metric.Evaluation.ranks_from_indices"]], "ranks_from_scores() (besskge.metric.evaluation method)": [[34, "besskge.metric.Evaluation.ranks_from_scores"]], "stacked_metrics_from_ranks() (besskge.metric.evaluation method)": [[34, "besskge.metric.Evaluation.stacked_metrics_from_ranks"]], "hitsatk (class in besskge.metric)": [[35, "besskge.metric.HitsAtK"]], "metrics_dict (in module besskge.metric)": [[36, "besskge.metric.METRICS_DICT"]], "reciprocalrank (class in besskge.metric)": [[37, "besskge.metric.ReciprocalRank"]], "besskge.negative_sampler": [[38, "module-besskge.negative_sampler"]], "placeholdernegativesampler (class in besskge.negative_sampler)": [[39, "besskge.negative_sampler.PlaceholderNegativeSampler"]], "corruption_scheme (besskge.negative_sampler.placeholdernegativesampler attribute)": [[39, "besskge.negative_sampler.PlaceholderNegativeSampler.corruption_scheme"]], "flat_negative_format (besskge.negative_sampler.placeholdernegativesampler attribute)": [[39, "besskge.negative_sampler.PlaceholderNegativeSampler.flat_negative_format"]], "local_sampling (besskge.negative_sampler.placeholdernegativesampler attribute)": [[39, "besskge.negative_sampler.PlaceholderNegativeSampler.local_sampling"]], "rng (besskge.negative_sampler.placeholdernegativesampler attribute)": [[39, "besskge.negative_sampler.PlaceholderNegativeSampler.rng"]], "randomshardednegativesampler (class in besskge.negative_sampler)": [[40, "besskge.negative_sampler.RandomShardedNegativeSampler"]], "corruption_scheme (besskge.negative_sampler.randomshardednegativesampler attribute)": [[40, "besskge.negative_sampler.RandomShardedNegativeSampler.corruption_scheme"]], "flat_negative_format (besskge.negative_sampler.randomshardednegativesampler attribute)": [[40, "besskge.negative_sampler.RandomShardedNegativeSampler.flat_negative_format"]], "local_sampling (besskge.negative_sampler.randomshardednegativesampler attribute)": [[40, "besskge.negative_sampler.RandomShardedNegativeSampler.local_sampling"]], "rng (besskge.negative_sampler.randomshardednegativesampler attribute)": [[40, "besskge.negative_sampler.RandomShardedNegativeSampler.rng"]], "shardednegativesampler (class in besskge.negative_sampler)": [[41, "besskge.negative_sampler.ShardedNegativeSampler"]], "corruption_scheme (besskge.negative_sampler.shardednegativesampler attribute)": [[41, "besskge.negative_sampler.ShardedNegativeSampler.corruption_scheme"]], "flat_negative_format (besskge.negative_sampler.shardednegativesampler attribute)": [[41, "besskge.negative_sampler.ShardedNegativeSampler.flat_negative_format"]], "local_sampling (besskge.negative_sampler.shardednegativesampler attribute)": [[41, "besskge.negative_sampler.ShardedNegativeSampler.local_sampling"]], "rng (besskge.negative_sampler.shardednegativesampler attribute)": [[41, "besskge.negative_sampler.ShardedNegativeSampler.rng"]], "triplebasedshardednegativesampler (class in besskge.negative_sampler)": [[42, "besskge.negative_sampler.TripleBasedShardedNegativeSampler"]], "corruption_scheme (besskge.negative_sampler.triplebasedshardednegativesampler attribute)": [[42, "besskge.negative_sampler.TripleBasedShardedNegativeSampler.corruption_scheme"]], "flat_negative_format (besskge.negative_sampler.triplebasedshardednegativesampler attribute)": [[42, "besskge.negative_sampler.TripleBasedShardedNegativeSampler.flat_negative_format"]], "local_sampling (besskge.negative_sampler.triplebasedshardednegativesampler attribute)": [[42, "besskge.negative_sampler.TripleBasedShardedNegativeSampler.local_sampling"]], "pad_negatives() (besskge.negative_sampler.triplebasedshardednegativesampler method)": [[42, "besskge.negative_sampler.TripleBasedShardedNegativeSampler.pad_negatives"]], "rng (besskge.negative_sampler.triplebasedshardednegativesampler attribute)": [[42, "besskge.negative_sampler.TripleBasedShardedNegativeSampler.rng"]], "shard_negatives() (besskge.negative_sampler.triplebasedshardednegativesampler method)": [[42, "besskge.negative_sampler.TripleBasedShardedNegativeSampler.shard_negatives"]], "typebasedshardednegativesampler (class in besskge.negative_sampler)": [[43, "besskge.negative_sampler.TypeBasedShardedNegativeSampler"]], "corruption_scheme (besskge.negative_sampler.typebasedshardednegativesampler attribute)": [[43, "besskge.negative_sampler.TypeBasedShardedNegativeSampler.corruption_scheme"]], "flat_negative_format (besskge.negative_sampler.typebasedshardednegativesampler attribute)": [[43, "besskge.negative_sampler.TypeBasedShardedNegativeSampler.flat_negative_format"]], "local_sampling (besskge.negative_sampler.typebasedshardednegativesampler attribute)": [[43, "besskge.negative_sampler.TypeBasedShardedNegativeSampler.local_sampling"]], "rng (besskge.negative_sampler.typebasedshardednegativesampler attribute)": [[43, "besskge.negative_sampler.TypeBasedShardedNegativeSampler.rng"]], "besskge.pipeline": [[44, "module-besskge.pipeline"]], "allscorespipeline (class in besskge.pipeline)": [[45, "besskge.pipeline.AllScoresPipeline"]], "forward() (besskge.pipeline.allscorespipeline method)": [[45, "besskge.pipeline.AllScoresPipeline.forward"]], "besskge.scoring": [[46, "module-besskge.scoring"]], "basescorefunction (class in besskge.scoring)": [[47, "besskge.scoring.BaseScoreFunction"]], "entity_embedding (besskge.scoring.basescorefunction attribute)": [[47, "besskge.scoring.BaseScoreFunction.entity_embedding"]], "forward() (besskge.scoring.basescorefunction method)": [[47, "besskge.scoring.BaseScoreFunction.forward"]], "negative_sample_sharing (besskge.scoring.basescorefunction attribute)": [[47, "besskge.scoring.BaseScoreFunction.negative_sample_sharing"]], "relation_embedding (besskge.scoring.basescorefunction attribute)": [[47, "besskge.scoring.BaseScoreFunction.relation_embedding"]], "score_heads() (besskge.scoring.basescorefunction method)": [[47, "besskge.scoring.BaseScoreFunction.score_heads"]], "score_tails() (besskge.scoring.basescorefunction method)": [[47, "besskge.scoring.BaseScoreFunction.score_tails"]], "score_triple() (besskge.scoring.basescorefunction method)": [[47, "besskge.scoring.BaseScoreFunction.score_triple"]], "sharding (besskge.scoring.basescorefunction attribute)": [[47, "besskge.scoring.BaseScoreFunction.sharding"]], "update_sharding() (besskge.scoring.basescorefunction method)": [[47, "besskge.scoring.BaseScoreFunction.update_sharding"]], "boxe (class in besskge.scoring)": [[48, "besskge.scoring.BoxE"]], "boxe_score() (besskge.scoring.boxe method)": [[48, "besskge.scoring.BoxE.boxe_score"]], "broadcasted_distance() (besskge.scoring.boxe method)": [[48, "besskge.scoring.BoxE.broadcasted_distance"]], "entity_embedding (besskge.scoring.boxe attribute)": [[48, "besskge.scoring.BoxE.entity_embedding"]], "forward() (besskge.scoring.boxe method)": [[48, "besskge.scoring.BoxE.forward"]], "negative_sample_sharing (besskge.scoring.boxe attribute)": [[48, "besskge.scoring.BoxE.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.boxe method)": [[48, "besskge.scoring.BoxE.reduce_embedding"]], "relation_embedding (besskge.scoring.boxe attribute)": [[48, "besskge.scoring.BoxE.relation_embedding"]], "score_heads() (besskge.scoring.boxe method)": [[48, "besskge.scoring.BoxE.score_heads"]], "score_tails() (besskge.scoring.boxe method)": [[48, "besskge.scoring.BoxE.score_tails"]], "score_triple() (besskge.scoring.boxe method)": [[48, "besskge.scoring.BoxE.score_triple"]], "sharding (besskge.scoring.boxe attribute)": [[48, "besskge.scoring.BoxE.sharding"]], "update_sharding() (besskge.scoring.boxe method)": [[48, "besskge.scoring.BoxE.update_sharding"]], "complex (class in besskge.scoring)": [[49, "besskge.scoring.ComplEx"]], "broadcasted_dot_product() (besskge.scoring.complex method)": [[49, "besskge.scoring.ComplEx.broadcasted_dot_product"]], "entity_embedding (besskge.scoring.complex attribute)": [[49, "besskge.scoring.ComplEx.entity_embedding"]], "forward() (besskge.scoring.complex method)": [[49, "besskge.scoring.ComplEx.forward"]], "negative_sample_sharing (besskge.scoring.complex attribute)": [[49, "besskge.scoring.ComplEx.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.complex method)": [[49, "besskge.scoring.ComplEx.reduce_embedding"]], "relation_embedding (besskge.scoring.complex attribute)": [[49, "besskge.scoring.ComplEx.relation_embedding"]], "score_heads() (besskge.scoring.complex method)": [[49, "besskge.scoring.ComplEx.score_heads"]], "score_tails() (besskge.scoring.complex method)": [[49, "besskge.scoring.ComplEx.score_tails"]], "score_triple() (besskge.scoring.complex method)": [[49, "besskge.scoring.ComplEx.score_triple"]], "sharding (besskge.scoring.complex attribute)": [[49, "besskge.scoring.ComplEx.sharding"]], "update_sharding() (besskge.scoring.complex method)": [[49, "besskge.scoring.ComplEx.update_sharding"]], "conve (class in besskge.scoring)": [[50, "besskge.scoring.ConvE"]], "broadcasted_dot_product() (besskge.scoring.conve method)": [[50, "besskge.scoring.ConvE.broadcasted_dot_product"]], "entity_embedding (besskge.scoring.conve attribute)": [[50, "besskge.scoring.ConvE.entity_embedding"]], "forward() (besskge.scoring.conve method)": [[50, "besskge.scoring.ConvE.forward"]], "negative_sample_sharing (besskge.scoring.conve attribute)": [[50, "besskge.scoring.ConvE.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.conve method)": [[50, "besskge.scoring.ConvE.reduce_embedding"]], "relation_embedding (besskge.scoring.conve attribute)": [[50, "besskge.scoring.ConvE.relation_embedding"]], "score_heads() (besskge.scoring.conve method)": [[50, "besskge.scoring.ConvE.score_heads"]], "score_tails() (besskge.scoring.conve method)": [[50, "besskge.scoring.ConvE.score_tails"]], "score_triple() (besskge.scoring.conve method)": [[50, "besskge.scoring.ConvE.score_triple"]], "sharding (besskge.scoring.conve attribute)": [[50, "besskge.scoring.ConvE.sharding"]], "update_sharding() (besskge.scoring.conve method)": [[50, "besskge.scoring.ConvE.update_sharding"]], "distmult (class in besskge.scoring)": [[51, "besskge.scoring.DistMult"]], "broadcasted_dot_product() (besskge.scoring.distmult method)": [[51, "besskge.scoring.DistMult.broadcasted_dot_product"]], "entity_embedding (besskge.scoring.distmult attribute)": [[51, "besskge.scoring.DistMult.entity_embedding"]], "forward() (besskge.scoring.distmult method)": [[51, "besskge.scoring.DistMult.forward"]], "negative_sample_sharing (besskge.scoring.distmult attribute)": [[51, "besskge.scoring.DistMult.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.distmult method)": [[51, "besskge.scoring.DistMult.reduce_embedding"]], "relation_embedding (besskge.scoring.distmult attribute)": [[51, "besskge.scoring.DistMult.relation_embedding"]], "score_heads() (besskge.scoring.distmult method)": [[51, "besskge.scoring.DistMult.score_heads"]], "score_tails() (besskge.scoring.distmult method)": [[51, "besskge.scoring.DistMult.score_tails"]], "score_triple() (besskge.scoring.distmult method)": [[51, "besskge.scoring.DistMult.score_triple"]], "sharding (besskge.scoring.distmult attribute)": [[51, "besskge.scoring.DistMult.sharding"]], "update_sharding() (besskge.scoring.distmult method)": [[51, "besskge.scoring.DistMult.update_sharding"]], "distancebasedscorefunction (class in besskge.scoring)": [[52, "besskge.scoring.DistanceBasedScoreFunction"]], "broadcasted_distance() (besskge.scoring.distancebasedscorefunction method)": [[52, "besskge.scoring.DistanceBasedScoreFunction.broadcasted_distance"]], "entity_embedding (besskge.scoring.distancebasedscorefunction attribute)": [[52, "besskge.scoring.DistanceBasedScoreFunction.entity_embedding"]], "forward() (besskge.scoring.distancebasedscorefunction method)": [[52, "besskge.scoring.DistanceBasedScoreFunction.forward"]], "negative_sample_sharing (besskge.scoring.distancebasedscorefunction attribute)": [[52, "besskge.scoring.DistanceBasedScoreFunction.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.distancebasedscorefunction method)": [[52, "besskge.scoring.DistanceBasedScoreFunction.reduce_embedding"]], "relation_embedding (besskge.scoring.distancebasedscorefunction attribute)": [[52, "besskge.scoring.DistanceBasedScoreFunction.relation_embedding"]], "score_heads() (besskge.scoring.distancebasedscorefunction method)": [[52, "besskge.scoring.DistanceBasedScoreFunction.score_heads"]], "score_tails() (besskge.scoring.distancebasedscorefunction method)": [[52, "besskge.scoring.DistanceBasedScoreFunction.score_tails"]], "score_triple() (besskge.scoring.distancebasedscorefunction method)": [[52, "besskge.scoring.DistanceBasedScoreFunction.score_triple"]], "sharding (besskge.scoring.distancebasedscorefunction attribute)": [[52, "besskge.scoring.DistanceBasedScoreFunction.sharding"]], "update_sharding() (besskge.scoring.distancebasedscorefunction method)": [[52, "besskge.scoring.DistanceBasedScoreFunction.update_sharding"]], "interht (class in besskge.scoring)": [[53, "besskge.scoring.InterHT"]], "broadcasted_distance() (besskge.scoring.interht method)": [[53, "besskge.scoring.InterHT.broadcasted_distance"]], "entity_embedding (besskge.scoring.interht attribute)": [[53, "besskge.scoring.InterHT.entity_embedding"]], "forward() (besskge.scoring.interht method)": [[53, "besskge.scoring.InterHT.forward"]], "negative_sample_sharing (besskge.scoring.interht attribute)": [[53, "besskge.scoring.InterHT.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.interht method)": [[53, "besskge.scoring.InterHT.reduce_embedding"]], "relation_embedding (besskge.scoring.interht attribute)": [[53, "besskge.scoring.InterHT.relation_embedding"]], "score_heads() (besskge.scoring.interht method)": [[53, "besskge.scoring.InterHT.score_heads"]], "score_tails() (besskge.scoring.interht method)": [[53, "besskge.scoring.InterHT.score_tails"]], "score_triple() (besskge.scoring.interht method)": [[53, "besskge.scoring.InterHT.score_triple"]], "sharding (besskge.scoring.interht attribute)": [[53, "besskge.scoring.InterHT.sharding"]], "update_sharding() (besskge.scoring.interht method)": [[53, "besskge.scoring.InterHT.update_sharding"]], "matrixdecompositionscorefunction (class in besskge.scoring)": [[54, "besskge.scoring.MatrixDecompositionScoreFunction"]], "broadcasted_dot_product() (besskge.scoring.matrixdecompositionscorefunction method)": [[54, "besskge.scoring.MatrixDecompositionScoreFunction.broadcasted_dot_product"]], "entity_embedding (besskge.scoring.matrixdecompositionscorefunction attribute)": [[54, "besskge.scoring.MatrixDecompositionScoreFunction.entity_embedding"]], "forward() (besskge.scoring.matrixdecompositionscorefunction method)": [[54, "besskge.scoring.MatrixDecompositionScoreFunction.forward"]], "negative_sample_sharing (besskge.scoring.matrixdecompositionscorefunction attribute)": [[54, "besskge.scoring.MatrixDecompositionScoreFunction.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.matrixdecompositionscorefunction method)": [[54, "besskge.scoring.MatrixDecompositionScoreFunction.reduce_embedding"]], "relation_embedding (besskge.scoring.matrixdecompositionscorefunction attribute)": [[54, "besskge.scoring.MatrixDecompositionScoreFunction.relation_embedding"]], "score_heads() (besskge.scoring.matrixdecompositionscorefunction method)": [[54, "besskge.scoring.MatrixDecompositionScoreFunction.score_heads"]], "score_tails() (besskge.scoring.matrixdecompositionscorefunction method)": [[54, "besskge.scoring.MatrixDecompositionScoreFunction.score_tails"]], "score_triple() (besskge.scoring.matrixdecompositionscorefunction method)": [[54, "besskge.scoring.MatrixDecompositionScoreFunction.score_triple"]], "sharding (besskge.scoring.matrixdecompositionscorefunction attribute)": [[54, "besskge.scoring.MatrixDecompositionScoreFunction.sharding"]], "update_sharding() (besskge.scoring.matrixdecompositionscorefunction method)": [[54, "besskge.scoring.MatrixDecompositionScoreFunction.update_sharding"]], "pairre (class in besskge.scoring)": [[55, "besskge.scoring.PairRE"]], "broadcasted_distance() (besskge.scoring.pairre method)": [[55, "besskge.scoring.PairRE.broadcasted_distance"]], "entity_embedding (besskge.scoring.pairre attribute)": [[55, "besskge.scoring.PairRE.entity_embedding"]], "forward() (besskge.scoring.pairre method)": [[55, "besskge.scoring.PairRE.forward"]], "negative_sample_sharing (besskge.scoring.pairre attribute)": [[55, "besskge.scoring.PairRE.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.pairre method)": [[55, "besskge.scoring.PairRE.reduce_embedding"]], "relation_embedding (besskge.scoring.pairre attribute)": [[55, "besskge.scoring.PairRE.relation_embedding"]], "score_heads() (besskge.scoring.pairre method)": [[55, "besskge.scoring.PairRE.score_heads"]], "score_tails() (besskge.scoring.pairre method)": [[55, "besskge.scoring.PairRE.score_tails"]], "score_triple() (besskge.scoring.pairre method)": [[55, "besskge.scoring.PairRE.score_triple"]], "sharding (besskge.scoring.pairre attribute)": [[55, "besskge.scoring.PairRE.sharding"]], "update_sharding() (besskge.scoring.pairre method)": [[55, "besskge.scoring.PairRE.update_sharding"]], "rotate (class in besskge.scoring)": [[56, "besskge.scoring.RotatE"]], "broadcasted_distance() (besskge.scoring.rotate method)": [[56, "besskge.scoring.RotatE.broadcasted_distance"]], "entity_embedding (besskge.scoring.rotate attribute)": [[56, "besskge.scoring.RotatE.entity_embedding"]], "forward() (besskge.scoring.rotate method)": [[56, "besskge.scoring.RotatE.forward"]], "negative_sample_sharing (besskge.scoring.rotate attribute)": [[56, "besskge.scoring.RotatE.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.rotate method)": [[56, "besskge.scoring.RotatE.reduce_embedding"]], "relation_embedding (besskge.scoring.rotate attribute)": [[56, "besskge.scoring.RotatE.relation_embedding"]], "score_heads() (besskge.scoring.rotate method)": [[56, "besskge.scoring.RotatE.score_heads"]], "score_tails() (besskge.scoring.rotate method)": [[56, "besskge.scoring.RotatE.score_tails"]], "score_triple() (besskge.scoring.rotate method)": [[56, "besskge.scoring.RotatE.score_triple"]], "sharding (besskge.scoring.rotate attribute)": [[56, "besskge.scoring.RotatE.sharding"]], "update_sharding() (besskge.scoring.rotate method)": [[56, "besskge.scoring.RotatE.update_sharding"]], "trans (class in besskge.scoring)": [[57, "besskge.scoring.TranS"]], "broadcasted_distance() (besskge.scoring.trans method)": [[57, "besskge.scoring.TranS.broadcasted_distance"]], "entity_embedding (besskge.scoring.trans attribute)": [[57, "besskge.scoring.TranS.entity_embedding"]], "forward() (besskge.scoring.trans method)": [[57, "besskge.scoring.TranS.forward"]], "negative_sample_sharing (besskge.scoring.trans attribute)": [[57, "besskge.scoring.TranS.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.trans method)": [[57, "besskge.scoring.TranS.reduce_embedding"]], "relation_embedding (besskge.scoring.trans attribute)": [[57, "besskge.scoring.TranS.relation_embedding"]], "score_heads() (besskge.scoring.trans method)": [[57, "besskge.scoring.TranS.score_heads"]], "score_tails() (besskge.scoring.trans method)": [[57, "besskge.scoring.TranS.score_tails"]], "score_triple() (besskge.scoring.trans method)": [[57, "besskge.scoring.TranS.score_triple"]], "sharding (besskge.scoring.trans attribute)": [[57, "besskge.scoring.TranS.sharding"]], "update_sharding() (besskge.scoring.trans method)": [[57, "besskge.scoring.TranS.update_sharding"]], "transe (class in besskge.scoring)": [[58, "besskge.scoring.TransE"]], "broadcasted_distance() (besskge.scoring.transe method)": [[58, "besskge.scoring.TransE.broadcasted_distance"]], "entity_embedding (besskge.scoring.transe attribute)": [[58, "besskge.scoring.TransE.entity_embedding"]], "forward() (besskge.scoring.transe method)": [[58, "besskge.scoring.TransE.forward"]], "negative_sample_sharing (besskge.scoring.transe attribute)": [[58, "besskge.scoring.TransE.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.transe method)": [[58, "besskge.scoring.TransE.reduce_embedding"]], "relation_embedding (besskge.scoring.transe attribute)": [[58, "besskge.scoring.TransE.relation_embedding"]], "score_heads() (besskge.scoring.transe method)": [[58, "besskge.scoring.TransE.score_heads"]], "score_tails() (besskge.scoring.transe method)": [[58, "besskge.scoring.TransE.score_tails"]], "score_triple() (besskge.scoring.transe method)": [[58, "besskge.scoring.TransE.score_triple"]], "sharding (besskge.scoring.transe attribute)": [[58, "besskge.scoring.TransE.sharding"]], "update_sharding() (besskge.scoring.transe method)": [[58, "besskge.scoring.TransE.update_sharding"]], "triplere (class in besskge.scoring)": [[59, "besskge.scoring.TripleRE"]], "broadcasted_distance() (besskge.scoring.triplere method)": [[59, "besskge.scoring.TripleRE.broadcasted_distance"]], "entity_embedding (besskge.scoring.triplere attribute)": [[59, "besskge.scoring.TripleRE.entity_embedding"]], "forward() (besskge.scoring.triplere method)": [[59, "besskge.scoring.TripleRE.forward"]], "negative_sample_sharing (besskge.scoring.triplere attribute)": [[59, "besskge.scoring.TripleRE.negative_sample_sharing"]], "reduce_embedding() (besskge.scoring.triplere method)": [[59, "besskge.scoring.TripleRE.reduce_embedding"]], "relation_embedding (besskge.scoring.triplere attribute)": [[59, "besskge.scoring.TripleRE.relation_embedding"]], "score_heads() (besskge.scoring.triplere method)": [[59, "besskge.scoring.TripleRE.score_heads"]], "score_tails() (besskge.scoring.triplere method)": [[59, "besskge.scoring.TripleRE.score_tails"]], "score_triple() (besskge.scoring.triplere method)": [[59, "besskge.scoring.TripleRE.score_triple"]], "sharding (besskge.scoring.triplere attribute)": [[59, "besskge.scoring.TripleRE.sharding"]], "update_sharding() (besskge.scoring.triplere method)": [[59, "besskge.scoring.TripleRE.update_sharding"]], "besskge.sharding": [[60, "module-besskge.sharding"]], "partitionedtripleset (class in besskge.sharding)": [[61, "besskge.sharding.PartitionedTripleSet"]], "create_from_dataset() (besskge.sharding.partitionedtripleset class method)": [[61, "besskge.sharding.PartitionedTripleSet.create_from_dataset"]], "create_from_queries() (besskge.sharding.partitionedtripleset class method)": [[61, "besskge.sharding.PartitionedTripleSet.create_from_queries"]], "dummy (besskge.sharding.partitionedtripleset attribute)": [[61, "besskge.sharding.PartitionedTripleSet.dummy"]], "inverse_triples (besskge.sharding.partitionedtripleset attribute)": [[61, "besskge.sharding.PartitionedTripleSet.inverse_triples"]], "neg_heads (besskge.sharding.partitionedtripleset attribute)": [[61, "besskge.sharding.PartitionedTripleSet.neg_heads"]], "neg_tails (besskge.sharding.partitionedtripleset attribute)": [[61, "besskge.sharding.PartitionedTripleSet.neg_tails"]], "partition_mode (besskge.sharding.partitionedtripleset attribute)": [[61, "besskge.sharding.PartitionedTripleSet.partition_mode"]], "sharding (besskge.sharding.partitionedtripleset attribute)": [[61, "besskge.sharding.PartitionedTripleSet.sharding"]], "triple_counts (besskge.sharding.partitionedtripleset attribute)": [[61, "besskge.sharding.PartitionedTripleSet.triple_counts"]], "triple_offsets (besskge.sharding.partitionedtripleset attribute)": [[61, "besskge.sharding.PartitionedTripleSet.triple_offsets"]], "triple_sort_idx (besskge.sharding.partitionedtripleset attribute)": [[61, "besskge.sharding.PartitionedTripleSet.triple_sort_idx"]], "triples (besskge.sharding.partitionedtripleset attribute)": [[61, "besskge.sharding.PartitionedTripleSet.triples"]], "types (besskge.sharding.partitionedtripleset attribute)": [[61, "besskge.sharding.PartitionedTripleSet.types"]], "sharding (class in besskge.sharding)": [[62, "besskge.sharding.Sharding"]], "create() (besskge.sharding.sharding class method)": [[62, "besskge.sharding.Sharding.create"]], "entity_to_idx (besskge.sharding.sharding attribute)": [[62, "besskge.sharding.Sharding.entity_to_idx"]], "entity_to_shard (besskge.sharding.sharding attribute)": [[62, "besskge.sharding.Sharding.entity_to_shard"]], "entity_type_counts (besskge.sharding.sharding attribute)": [[62, "besskge.sharding.Sharding.entity_type_counts"]], "entity_type_offsets (besskge.sharding.sharding attribute)": [[62, "besskge.sharding.Sharding.entity_type_offsets"]], "load() (besskge.sharding.sharding class method)": [[62, "besskge.sharding.Sharding.load"]], "max_entity_per_shard (besskge.sharding.sharding property)": [[62, "besskge.sharding.Sharding.max_entity_per_shard"]], "n_entity (besskge.sharding.sharding property)": [[62, "besskge.sharding.Sharding.n_entity"]], "n_shard (besskge.sharding.sharding attribute)": [[62, "besskge.sharding.Sharding.n_shard"]], "save() (besskge.sharding.sharding method)": [[62, "besskge.sharding.Sharding.save"]], "shard_and_idx_to_entity (besskge.sharding.sharding attribute)": [[62, "besskge.sharding.Sharding.shard_and_idx_to_entity"]], "shard_counts (besskge.sharding.sharding attribute)": [[62, "besskge.sharding.Sharding.shard_counts"]], "besskge.utils": [[63, "module-besskge.utils"]], "complex_multiplication() (in module besskge.utils)": [[64, "besskge.utils.complex_multiplication"]], "complex_rotation() (in module besskge.utils)": [[65, "besskge.utils.complex_rotation"]], "gather_indices() (in module besskge.utils)": [[66, "besskge.utils.gather_indices"]], "get_entity_filter() (in module besskge.utils)": [[67, "besskge.utils.get_entity_filter"]], "besskge": [[68, "module-besskge"]]}})
\ No newline at end of file