mirror of
https://github.com/samsledje/D-SCRIPT.git
synced 2026-06-04 23:14:22 +08:00
346 lines
18 KiB
HTML
346 lines
18 KiB
HTML
|
||
|
||
<!DOCTYPE html>
|
||
<html class="writer-html5" lang="en" >
|
||
<head>
|
||
<meta charset="utf-8" />
|
||
|
||
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
|
||
|
||
<title>dscript.commands — D-SCRIPT v1.0-beta documentation</title>
|
||
|
||
|
||
|
||
<link rel="stylesheet" href="../_static/css/theme.css" type="text/css" />
|
||
<link rel="stylesheet" href="../_static/pygments.css" type="text/css" />
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
<!--[if lt IE 9]>
|
||
<script src="../_static/js/html5shiv.min.js"></script>
|
||
<![endif]-->
|
||
|
||
|
||
<script type="text/javascript" id="documentation_options" data-url_root="../" src="../_static/documentation_options.js"></script>
|
||
<script src="../_static/jquery.js"></script>
|
||
<script src="../_static/underscore.js"></script>
|
||
<script src="../_static/doctools.js"></script>
|
||
<script async="async" src="https://cdnjs.cloudflare.com/ajax/libs/mathjax/2.7.7/latest.js?config=TeX-AMS-MML_HTMLorMML"></script>
|
||
|
||
<script type="text/javascript" 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="dscript.models" href="dscript.models.html" />
|
||
<link rel="prev" title="API" href="index.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"> D-SCRIPT
|
||
|
||
|
||
|
||
</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" />
|
||
<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="main navigation">
|
||
|
||
|
||
|
||
|
||
|
||
|
||
<ul class="current">
|
||
<li class="toctree-l1"><a class="reference internal" href="../installation.html">Installation</a></li>
|
||
<li class="toctree-l1"><a class="reference internal" href="../usage.html">Usage</a></li>
|
||
<li class="toctree-l1"><a class="reference internal" href="../data.html">Data</a></li>
|
||
<li class="toctree-l1 current"><a class="reference internal" href="index.html">API</a><ul class="current">
|
||
<li class="toctree-l2 current"><a class="current reference internal" href="#">dscript.commands</a><ul>
|
||
<li class="toctree-l3"><a class="reference internal" href="#dscript-commands-predict">dscript.commands.predict</a></li>
|
||
<li class="toctree-l3"><a class="reference internal" href="#dscript-commands-embed">dscript.commands.embed</a></li>
|
||
<li class="toctree-l3"><a class="reference internal" href="#dscript-commands-train">dscript.commands.train</a></li>
|
||
<li class="toctree-l3"><a class="reference internal" href="#dscript-commands-eval">dscript.commands.eval</a></li>
|
||
</ul>
|
||
</li>
|
||
<li class="toctree-l2"><a class="reference internal" href="dscript.models.html">dscript.models</a></li>
|
||
<li class="toctree-l2"><a class="reference internal" href="index.html#module-dscript.alphabets">dscript.alphabets</a></li>
|
||
<li class="toctree-l2"><a class="reference internal" href="index.html#module-dscript.fasta">dscript.fasta</a></li>
|
||
<li class="toctree-l2"><a class="reference internal" href="index.html#module-dscript.language_model">dscript.language_model</a></li>
|
||
<li class="toctree-l2"><a class="reference internal" href="index.html#module-dscript.pretrained">dscript.pretrained</a></li>
|
||
<li class="toctree-l2"><a class="reference internal" href="index.html#module-dscript.utils">dscript.utils</a></li>
|
||
</ul>
|
||
</li>
|
||
</ul>
|
||
|
||
|
||
|
||
</div>
|
||
|
||
</div>
|
||
</nav>
|
||
|
||
<section data-toggle="wy-nav-shift" class="wy-nav-content-wrap">
|
||
|
||
|
||
<nav class="wy-nav-top" aria-label="top navigation">
|
||
|
||
<i data-toggle="wy-nav-top" class="fa fa-bars"></i>
|
||
<a href="../index.html">D-SCRIPT</a>
|
||
|
||
</nav>
|
||
|
||
|
||
<div class="wy-nav-content">
|
||
|
||
<div class="rst-content">
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
<div role="navigation" aria-label="breadcrumbs navigation">
|
||
|
||
<ul class="wy-breadcrumbs">
|
||
|
||
<li><a href="../index.html" class="icon icon-home"></a> »</li>
|
||
|
||
<li><a href="index.html">API</a> »</li>
|
||
|
||
<li>dscript.commands</li>
|
||
|
||
|
||
<li class="wy-breadcrumbs-aside">
|
||
|
||
|
||
<a href="../_sources/api/dscript.commands.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">
|
||
|
||
<div class="section" id="dscript-commands">
|
||
<h1>dscript.commands<a class="headerlink" href="#dscript-commands" title="Permalink to this headline">¶</a></h1>
|
||
<div class="section" id="dscript-commands-predict">
|
||
<h2>dscript.commands.predict<a class="headerlink" href="#dscript-commands-predict" title="Permalink to this headline">¶</a></h2>
|
||
<p>See <a class="reference external" href="../usage.html#prediction">Prediction</a> for full usage details.</p>
|
||
<span class="target" id="module-dscript.commands.predict"></span><p>Make new predictions with a pre-trained model. One of –seqs or –embeddings is required.</p>
|
||
</div>
|
||
<div class="section" id="dscript-commands-embed">
|
||
<h2>dscript.commands.embed<a class="headerlink" href="#dscript-commands-embed" title="Permalink to this headline">¶</a></h2>
|
||
<p>See <a class="reference external" href="../usage.html#embedding">Embedding</a> for full usage details.</p>
|
||
<span class="target" id="module-dscript.commands.embed"></span><p>Generate new embeddings using pre-trained language model.</p>
|
||
</div>
|
||
<div class="section" id="dscript-commands-train">
|
||
<h2>dscript.commands.train<a class="headerlink" href="#dscript-commands-train" title="Permalink to this headline">¶</a></h2>
|
||
<p>See <a class="reference external" href="../usage.html#training">Training</a> for full usage details.</p>
|
||
<span class="target" id="module-dscript.commands.train"></span><p>Train a new model.</p>
|
||
<dl class="py function">
|
||
<dt id="dscript.commands.train.interaction_eval">
|
||
<code class="sig-prename descclassname"><span class="pre">dscript.commands.train.</span></code><code class="sig-name descname"><span class="pre">interaction_eval</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">model</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">test_iterator</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">tensors</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">use_cuda</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/dscript/commands/train.html#interaction_eval"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#dscript.commands.train.interaction_eval" title="Permalink to this definition">¶</a></dt>
|
||
<dd><p>Evaluate test data set performance.</p>
|
||
<dl class="field-list simple">
|
||
<dt class="field-odd">Parameters</dt>
|
||
<dd class="field-odd"><ul class="simple">
|
||
<li><p><strong>model</strong> (<a class="reference internal" href="dscript.models.html#dscript.models.interaction.ModelInteraction" title="dscript.models.interaction.ModelInteraction"><em>dscript.models.interaction.ModelInteraction</em></a>) – Model to be trained</p></li>
|
||
<li><p><strong>test_iterator</strong> (<em>torch.utils.data.DataLoader</em>) – Test data iterator</p></li>
|
||
<li><p><strong>tensors</strong> (<em>dict</em><em>[</em><em>str</em><em>, </em><em>torch.Tensor</em><em>]</em>) – Dictionary of protein names to embeddings</p></li>
|
||
<li><p><strong>use_cuda</strong> (<em>bool</em>) – Whether to use GPU</p></li>
|
||
</ul>
|
||
</dd>
|
||
<dt class="field-even">Returns</dt>
|
||
<dd class="field-even"><p>(Loss, number correct, mean square error, precision, recall, F1 Score, AUPR)</p>
|
||
</dd>
|
||
<dt class="field-odd">Return type</dt>
|
||
<dd class="field-odd"><p>(torch.Tensor, int, torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor)</p>
|
||
</dd>
|
||
</dl>
|
||
</dd></dl>
|
||
|
||
<dl class="py function">
|
||
<dt id="dscript.commands.train.interaction_grad">
|
||
<code class="sig-prename descclassname"><span class="pre">dscript.commands.train.</span></code><code class="sig-name descname"><span class="pre">interaction_grad</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">model</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">n0</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">n1</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">y</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">tensors</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">use_cuda</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">weight</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">0.35</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/dscript/commands/train.html#interaction_grad"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#dscript.commands.train.interaction_grad" title="Permalink to this definition">¶</a></dt>
|
||
<dd><p>Compute gradient and backpropagate loss for a batch.</p>
|
||
<dl class="field-list simple">
|
||
<dt class="field-odd">Parameters</dt>
|
||
<dd class="field-odd"><ul class="simple">
|
||
<li><p><strong>model</strong> (<a class="reference internal" href="dscript.models.html#dscript.models.interaction.ModelInteraction" title="dscript.models.interaction.ModelInteraction"><em>dscript.models.interaction.ModelInteraction</em></a>) – Model to be trained</p></li>
|
||
<li><p><strong>n0</strong> (<em>list</em><em>[</em><em>str</em><em>]</em>) – First protein names</p></li>
|
||
<li><p><strong>n1</strong> (<em>list</em><em>[</em><em>str</em><em>]</em>) – Second protein names</p></li>
|
||
<li><p><strong>y</strong> (<em>torch.Tensor</em>) – Interaction labels</p></li>
|
||
<li><p><strong>tensors</strong> (<em>dict</em><em>[</em><em>str</em><em>, </em><em>torch.Tensor</em><em>]</em>) – Dictionary of protein names to embeddings</p></li>
|
||
<li><p><strong>use_cuda</strong> (<em>bool</em>) – Whether to use GPU</p></li>
|
||
<li><p><strong>weight</strong> (<em>float</em>) – Weight on the contact map magnitude objective. BCE loss is <span class="math notranslate nohighlight">\(1 - \text{weight}\)</span>.</p></li>
|
||
</ul>
|
||
</dd>
|
||
<dt class="field-even">Returns</dt>
|
||
<dd class="field-even"><p>(Loss, number correct, mean square error, batch size)</p>
|
||
</dd>
|
||
<dt class="field-odd">Return type</dt>
|
||
<dd class="field-odd"><p>(torch.Tensor, int, torch.Tensor, int)</p>
|
||
</dd>
|
||
</dl>
|
||
</dd></dl>
|
||
|
||
<dl class="py function">
|
||
<dt id="dscript.commands.train.predict_cmap_interaction">
|
||
<code class="sig-prename descclassname"><span class="pre">dscript.commands.train.</span></code><code class="sig-name descname"><span class="pre">predict_cmap_interaction</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">model</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">n0</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">n1</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">tensors</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">use_cuda</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/dscript/commands/train.html#predict_cmap_interaction"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#dscript.commands.train.predict_cmap_interaction" title="Permalink to this definition">¶</a></dt>
|
||
<dd><p>Predict whether a list of protein pairs will interact, as well as their contact map.</p>
|
||
<dl class="field-list simple">
|
||
<dt class="field-odd">Parameters</dt>
|
||
<dd class="field-odd"><ul class="simple">
|
||
<li><p><strong>model</strong> (<a class="reference internal" href="dscript.models.html#dscript.models.interaction.ModelInteraction" title="dscript.models.interaction.ModelInteraction"><em>dscript.models.interaction.ModelInteraction</em></a>) – Model to be trained</p></li>
|
||
<li><p><strong>n0</strong> (<em>list</em><em>[</em><em>str</em><em>]</em>) – First protein names</p></li>
|
||
<li><p><strong>n1</strong> (<em>list</em><em>[</em><em>str</em><em>]</em>) – Second protein names</p></li>
|
||
<li><p><strong>tensors</strong> (<em>dict</em><em>[</em><em>str</em><em>, </em><em>torch.Tensor</em><em>]</em>) – Dictionary of protein names to embeddings</p></li>
|
||
<li><p><strong>use_cuda</strong> (<em>bool</em>) – Whether to use GPU</p></li>
|
||
</ul>
|
||
</dd>
|
||
</dl>
|
||
</dd></dl>
|
||
|
||
<dl class="py function">
|
||
<dt id="dscript.commands.train.predict_interaction">
|
||
<code class="sig-prename descclassname"><span class="pre">dscript.commands.train.</span></code><code class="sig-name descname"><span class="pre">predict_interaction</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">model</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">n0</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">n1</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">tensors</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">use_cuda</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/dscript/commands/train.html#predict_interaction"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#dscript.commands.train.predict_interaction" title="Permalink to this definition">¶</a></dt>
|
||
<dd><p>Predict whether a list of protein pairs will interact.</p>
|
||
<dl class="field-list simple">
|
||
<dt class="field-odd">Parameters</dt>
|
||
<dd class="field-odd"><ul class="simple">
|
||
<li><p><strong>model</strong> (<a class="reference internal" href="dscript.models.html#dscript.models.interaction.ModelInteraction" title="dscript.models.interaction.ModelInteraction"><em>dscript.models.interaction.ModelInteraction</em></a>) – Model to be trained</p></li>
|
||
<li><p><strong>n0</strong> (<em>list</em><em>[</em><em>str</em><em>]</em>) – First protein names</p></li>
|
||
<li><p><strong>n1</strong> (<em>list</em><em>[</em><em>str</em><em>]</em>) – Second protein names</p></li>
|
||
<li><p><strong>tensors</strong> (<em>dict</em><em>[</em><em>str</em><em>, </em><em>torch.Tensor</em><em>]</em>) – Dictionary of protein names to embeddings</p></li>
|
||
<li><p><strong>use_cuda</strong> (<em>bool</em>) – Whether to use GPU</p></li>
|
||
</ul>
|
||
</dd>
|
||
</dl>
|
||
</dd></dl>
|
||
|
||
</div>
|
||
<div class="section" id="dscript-commands-eval">
|
||
<h2>dscript.commands.eval<a class="headerlink" href="#dscript-commands-eval" title="Permalink to this headline">¶</a></h2>
|
||
<p>See <a class="reference external" href="../usage.html#evaluation">Evaluation</a> for full usage details.</p>
|
||
<span class="target" id="module-dscript.commands.eval"></span><p>Evaluate a trained model.</p>
|
||
<dl class="py function">
|
||
<dt id="dscript.commands.eval.plot_eval_predictions">
|
||
<code class="sig-prename descclassname"><span class="pre">dscript.commands.eval.</span></code><code class="sig-name descname"><span class="pre">plot_eval_predictions</span></code><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">labels</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">predictions</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">path</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'figure'</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/dscript/commands/eval.html#plot_eval_predictions"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#dscript.commands.eval.plot_eval_predictions" title="Permalink to this definition">¶</a></dt>
|
||
<dd><p>Plot histogram of positive and negative predictions, precision-recall curve, and receiver operating characteristic curve.</p>
|
||
<dl class="field-list simple">
|
||
<dt class="field-odd">Parameters</dt>
|
||
<dd class="field-odd"><ul class="simple">
|
||
<li><p><strong>y</strong> (<em>np.ndarray</em>) – Labels</p></li>
|
||
<li><p><strong>phat</strong> (<em>np.ndarray</em>) – Predicted probabilities</p></li>
|
||
<li><p><strong>path</strong> (<em>str</em>) – File prefix for plots to be saved to [default: figure]</p></li>
|
||
</ul>
|
||
</dd>
|
||
</dl>
|
||
</dd></dl>
|
||
|
||
</div>
|
||
</div>
|
||
|
||
|
||
</div>
|
||
|
||
</div>
|
||
<footer>
|
||
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation">
|
||
<a href="dscript.models.html" class="btn btn-neutral float-right" title="dscript.models" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
|
||
<a href="index.html" class="btn btn-neutral float-left" title="API" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> Previous</a>
|
||
</div>
|
||
|
||
<hr/>
|
||
|
||
<div role="contentinfo">
|
||
<p>
|
||
© Copyright 2020, Samuel Sledzieski, Rohit Singh.
|
||
|
||
</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 type="text/javascript">
|
||
jQuery(function () {
|
||
SphinxRtdTheme.Navigation.enable(true);
|
||
});
|
||
</script>
|
||
|
||
|
||
|
||
|
||
|
||
|
||
</body>
|
||
</html>
|