|
197 | 197 | <input type="hidden" name="area" value="default">
|
198 | 198 | </form>
|
199 | 199 | <div id="searchbox"></div><div class="sidebar-scroll"><div class="sidebar-tree">
|
200 |
| - <p class="caption" role="heading"><span class="caption-text">Contents:</span></p> |
| 200 | + <p class="caption" role="heading"><span class="caption-text">Getting Started:</span></p> |
201 | 201 | <ul>
|
202 |
| -<li class="toctree-l1"><a class="reference internal" href="../../../api.html">API Reference</a></li> |
| 202 | +<li class="toctree-l1"><a class="reference internal" href="../../../install.html">Installation</a></li> |
| 203 | +<li class="toctree-l1"><a class="reference internal" href="../../../example.html">Examples</a></li> |
| 204 | +</ul> |
| 205 | +<p class="caption" role="heading"><span class="caption-text">API Documentation:</span></p> |
| 206 | +<ul> |
| 207 | +<li class="toctree-l1 has-children"><a class="reference internal" href="../../../api/index.html">API Reference</a><input class="toctree-checkbox" id="toctree-checkbox-1" name="toctree-checkbox-1" role="switch" type="checkbox"/><label for="toctree-checkbox-1"><div class="visually-hidden">Toggle navigation of API Reference</div><i class="icon"><svg><use href="#svg-arrow-right"></use></svg></i></label><ul> |
| 208 | +<li class="toctree-l2"><a class="reference internal" href="../../../api/predictor.html">API Reference for Predictors</a></li> |
| 209 | +<li class="toctree-l2"><a class="reference internal" href="../../../api/generator.html">API Reference for Generators</a></li> |
| 210 | +<li class="toctree-l2"><a class="reference internal" href="../../../api/encoder.html">API Reference for Encoders</a></li> |
| 211 | +<li class="toctree-l2"><a class="reference internal" href="../../../api/utils.html">API Reference for Utilities</a></li> |
| 212 | +<li class="toctree-l2"><a class="reference internal" href="../../../api/nn.html">API Reference for Neural Network Components</a></li> |
| 213 | +<li class="toctree-l2"><a class="reference internal" href="../../../api/base.html">API Reference for Base Modules</a></li> |
| 214 | +</ul> |
| 215 | +</li> |
| 216 | +</ul> |
| 217 | +<p class="caption" role="heading"><span class="caption-text">Additional Information:</span></p> |
| 218 | +<ul> |
| 219 | +<li class="toctree-l1"><a class="reference internal" href="../../../additional.html">Additional Information</a></li> |
203 | 220 | </ul>
|
204 | 221 |
|
205 | 222 | </div>
|
@@ -244,22 +261,22 @@ <h1>Source code for torch_molecule.base.encoder</h1><div class="highlight"><pre>
|
244 | 261 | <span class="kn">from</span><span class="w"> </span><span class="nn">.base</span><span class="w"> </span><span class="kn">import</span> <span class="n">BaseModel</span>
|
245 | 262 |
|
246 | 263 | <div class="viewcode-block" id="BaseMolecularEncoder">
|
247 |
| -<a class="viewcode-back" href="../../../api.html#torch_molecule.base.BaseMolecularEncoder">[docs]</a> |
| 264 | +<a class="viewcode-back" href="../../../api/utils.html#torch_molecule.base.BaseMolecularEncoder">[docs]</a> |
248 | 265 | <span class="nd">@dataclass</span>
|
249 | 266 | <span class="k">class</span><span class="w"> </span><span class="nc">BaseMolecularEncoder</span><span class="p">(</span><span class="n">BaseModel</span><span class="p">,</span> <span class="n">ABC</span><span class="p">):</span>
|
250 | 267 | <span class="w"> </span><span class="sd">"""Base class for molecular representation learning."""</span>
|
251 | 268 |
|
252 | 269 | <span class="n">model_name</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="n">field</span><span class="p">(</span><span class="n">default</span><span class="o">=</span><span class="s2">"BaseMolecularEncoder"</span><span class="p">)</span>
|
253 | 270 |
|
254 | 271 | <div class="viewcode-block" id="BaseMolecularEncoder.encode">
|
255 |
| -<a class="viewcode-back" href="../../../api.html#torch_molecule.base.BaseMolecularEncoder.encode">[docs]</a> |
| 272 | +<a class="viewcode-back" href="../../../api/utils.html#torch_molecule.base.BaseMolecularEncoder.encode">[docs]</a> |
256 | 273 | <span class="nd">@abstractmethod</span>
|
257 | 274 | <span class="k">def</span><span class="w"> </span><span class="nf">encode</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">X</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="nb">str</span><span class="p">],</span> <span class="n">return_type</span><span class="p">:</span> <span class="n">Literal</span><span class="p">[</span><span class="s2">"np"</span><span class="p">,</span> <span class="s2">"pt"</span><span class="p">]</span> <span class="o">=</span> <span class="s2">"pt"</span><span class="p">)</span> <span class="o">-></span> <span class="n">Union</span><span class="p">[</span><span class="n">np</span><span class="o">.</span><span class="n">ndarray</span><span class="p">,</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">]:</span>
|
258 | 275 | <span class="k">pass</span></div>
|
259 | 276 |
|
260 | 277 |
|
261 | 278 | <div class="viewcode-block" id="BaseMolecularEncoder.fit">
|
262 |
| -<a class="viewcode-back" href="../../../api.html#torch_molecule.base.BaseMolecularEncoder.fit">[docs]</a> |
| 279 | +<a class="viewcode-back" href="../../../api/utils.html#torch_molecule.base.BaseMolecularEncoder.fit">[docs]</a> |
263 | 280 | <span class="nd">@abstractmethod</span>
|
264 | 281 | <span class="k">def</span><span class="w"> </span><span class="nf">fit</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">X</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="nb">str</span><span class="p">],</span> <span class="n">y</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">np</span><span class="o">.</span><span class="n">ndarray</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">)</span> <span class="o">-></span> <span class="s2">"BaseMolecularEncoder"</span><span class="p">:</span>
|
265 | 282 | <span class="k">pass</span></div>
|
|
0 commit comments