|
a |
|
b/docs/build/html/notes/configuration.html |
|
|
1 |
|
|
|
2 |
|
|
|
3 |
<!DOCTYPE html> |
|
|
4 |
<!--[if IE 8]><html class="no-js lt-ie9" lang="en" > <![endif]--> |
|
|
5 |
<!--[if gt IE 8]><!--> <html class="no-js" lang="en" > <!--<![endif]--> |
|
|
6 |
<head> |
|
|
7 |
<meta charset="utf-8"> |
|
|
8 |
|
|
|
9 |
<meta name="viewport" content="width=device-width, initial-scale=1.0"> |
|
|
10 |
|
|
|
11 |
<title>Configuration — DeepPurpose 0.0.1 documentation</title> |
|
|
12 |
|
|
|
13 |
|
|
|
14 |
|
|
|
15 |
|
|
|
16 |
|
|
|
17 |
|
|
|
18 |
|
|
|
19 |
|
|
|
20 |
<script type="text/javascript" src="../_static/js/modernizr.min.js"></script> |
|
|
21 |
|
|
|
22 |
|
|
|
23 |
<script type="text/javascript" id="documentation_options" data-url_root="../" src="../_static/documentation_options.js"></script> |
|
|
24 |
<script src="../_static/jquery.js"></script> |
|
|
25 |
<script src="../_static/underscore.js"></script> |
|
|
26 |
<script src="../_static/doctools.js"></script> |
|
|
27 |
<script src="../_static/language_data.js"></script> |
|
|
28 |
|
|
|
29 |
<script type="text/javascript" src="../_static/js/theme.js"></script> |
|
|
30 |
|
|
|
31 |
|
|
|
32 |
|
|
|
33 |
|
|
|
34 |
<link rel="stylesheet" href="../_static/css/theme.css" type="text/css" /> |
|
|
35 |
<link rel="stylesheet" href="../_static/pygments.css" type="text/css" /> |
|
|
36 |
<link rel="stylesheet" href="https://fonts.googleapis.com/css?family=Raleway" type="text/css" /> |
|
|
37 |
<link rel="stylesheet" href="../../build/html/_static/css/deeppurpose_docs_theme.css" type="text/css" /> |
|
|
38 |
<link rel="index" title="Index" href="../genindex.html" /> |
|
|
39 |
<link rel="search" title="Search" href="../search.html" /> |
|
|
40 |
<link rel="next" title="Utility Function" href="utility_function.html" /> |
|
|
41 |
<link rel="prev" title="Processing Data" href="process_data.html" /> |
|
|
42 |
</head> |
|
|
43 |
|
|
|
44 |
<body class="wy-body-for-nav"> |
|
|
45 |
|
|
|
46 |
|
|
|
47 |
<div class="wy-grid-for-nav"> |
|
|
48 |
|
|
|
49 |
<nav data-toggle="wy-nav-shift" class="wy-nav-side"> |
|
|
50 |
<div class="wy-side-scroll"> |
|
|
51 |
<div class="wy-side-nav-search" > |
|
|
52 |
|
|
|
53 |
|
|
|
54 |
|
|
|
55 |
<a href="../index.html" class="icon icon-home"> DeepPurpose |
|
|
56 |
|
|
|
57 |
|
|
|
58 |
|
|
|
59 |
|
|
|
60 |
<img src="../_static/logo_dp_2.png" class="logo" alt="Logo"/> |
|
|
61 |
|
|
|
62 |
</a> |
|
|
63 |
|
|
|
64 |
|
|
|
65 |
|
|
|
66 |
|
|
|
67 |
|
|
|
68 |
|
|
|
69 |
|
|
|
70 |
<div role="search"> |
|
|
71 |
<form id="rtd-search-form" class="wy-form" action="../search.html" method="get"> |
|
|
72 |
<input type="text" name="q" placeholder="Search docs" /> |
|
|
73 |
<input type="hidden" name="check_keywords" value="yes" /> |
|
|
74 |
<input type="hidden" name="area" value="default" /> |
|
|
75 |
</form> |
|
|
76 |
</div> |
|
|
77 |
|
|
|
78 |
|
|
|
79 |
</div> |
|
|
80 |
|
|
|
81 |
<div class="wy-menu wy-menu-vertical" data-spy="affix" role="navigation" aria-label="main navigation"> |
|
|
82 |
|
|
|
83 |
|
|
|
84 |
|
|
|
85 |
|
|
|
86 |
|
|
|
87 |
|
|
|
88 |
<p class="caption"><span class="caption-text">Background</span></p> |
|
|
89 |
<ul> |
|
|
90 |
<li class="toctree-l1"><a class="reference internal" href="introduction.html">Features of DeepPurpose</a><ul> |
|
|
91 |
<li class="toctree-l2"><a class="reference internal" href="introduction.html#features">Features</a></li> |
|
|
92 |
</ul> |
|
|
93 |
</li> |
|
|
94 |
<li class="toctree-l1"><a class="reference internal" href="DTI.html">What is drug repurposing, virtual screening and drug-target interaction prediction?</a><ul> |
|
|
95 |
<li class="toctree-l2"><a class="reference internal" href="DTI.html#drug-repurposing">Drug Repurposing</a></li> |
|
|
96 |
<li class="toctree-l2"><a class="reference internal" href="DTI.html#virtual-screening">Virtual Screening</a></li> |
|
|
97 |
<li class="toctree-l2"><a class="reference internal" href="DTI.html#drug-target-interaction">Drug-Target Interaction</a></li> |
|
|
98 |
</ul> |
|
|
99 |
</li> |
|
|
100 |
</ul> |
|
|
101 |
<p class="caption"><span class="caption-text">How to run</span></p> |
|
|
102 |
<ul> |
|
|
103 |
<li class="toctree-l1"><a class="reference internal" href="download.html">Download Code & Install</a><ul> |
|
|
104 |
<li class="toctree-l2"><a class="reference internal" href="download.html#download-code">Download Code</a></li> |
|
|
105 |
<li class="toctree-l2"><a class="reference internal" href="download.html#first-time-usage-setup-conda-environment">First time usage: setup conda environment</a></li> |
|
|
106 |
<li class="toctree-l2"><a class="reference internal" href="download.html#second-time-and-later">Second time and later</a></li> |
|
|
107 |
</ul> |
|
|
108 |
</li> |
|
|
109 |
<li class="toctree-l1"><a class="reference internal" href="casestudy.html">Case Study</a></li> |
|
|
110 |
</ul> |
|
|
111 |
<p class="caption"><span class="caption-text">Package Reference</span></p> |
|
|
112 |
<ul> |
|
|
113 |
<li class="toctree-l1"><a class="reference internal" href="models.html">DeepPurpose.models</a><ul> |
|
|
114 |
<li class="toctree-l2"><a class="reference internal" href="dtba/classifier.html">Classifier</a></li> |
|
|
115 |
<li class="toctree-l2"><a class="reference internal" href="dtba/dbta.html">Drug Target Binding Affinity (DTBA) Model</a></li> |
|
|
116 |
<li class="toctree-l2"><a class="reference internal" href="encoders/transformer.html">Transformer</a></li> |
|
|
117 |
<li class="toctree-l2"><a class="reference internal" href="encoders/mpnn.html">Message Passing Neural Network (MPNN)</a></li> |
|
|
118 |
<li class="toctree-l2"><a class="reference internal" href="encoders/cnnrnn.html">CNN+RNN</a></li> |
|
|
119 |
<li class="toctree-l2"><a class="reference internal" href="encoders/cnn.html">CNN</a></li> |
|
|
120 |
<li class="toctree-l2"><a class="reference internal" href="encoders/mlp.html">MLP</a></li> |
|
|
121 |
</ul> |
|
|
122 |
</li> |
|
|
123 |
<li class="toctree-l1"><a class="reference internal" href="dataset.html">DeepPurpose.dataset</a><ul> |
|
|
124 |
<li class="toctree-l2"><a class="reference internal" href="data/read_file_training_dataset_bioassay.html">read_file_training_dataset_bioassay</a></li> |
|
|
125 |
<li class="toctree-l2"><a class="reference internal" href="data/read_file_training_dataset_drug_target_pairs.html">read_file_training_dataset_drug_target_pairs</a></li> |
|
|
126 |
<li class="toctree-l2"><a class="reference internal" href="data/read_file_virtual_screening_drug_target_pairs.html">read_file_virtual_screening_drug_target_pairs</a></li> |
|
|
127 |
<li class="toctree-l2"><a class="reference internal" href="data/read_file_repurposing_library.html">load bioarray dataset (read_file_training_dataset_bioassay)</a></li> |
|
|
128 |
<li class="toctree-l2"><a class="reference internal" href="data/read_file_target_sequence.html">read_file_target_sequence</a></li> |
|
|
129 |
<li class="toctree-l2"><a class="reference internal" href="data/download_BindingDB.html">download_DrugTargetCommons</a></li> |
|
|
130 |
<li class="toctree-l2"><a class="reference internal" href="data/process_BindingDB.html">process_BindingDB</a></li> |
|
|
131 |
<li class="toctree-l2"><a class="reference internal" href="data/load_process_DAVIS.html">load_process_DAVIS</a></li> |
|
|
132 |
<li class="toctree-l2"><a class="reference internal" href="data/load_process_KIBA.html">load_process_KIBA</a></li> |
|
|
133 |
<li class="toctree-l2"><a class="reference internal" href="data/load_AID1706_txt_file.html">load_AID1706_txt_file</a></li> |
|
|
134 |
<li class="toctree-l2"><a class="reference internal" href="data/load_AID1706_SARS_CoV_3CL.html">load_AID1706_SARS_CoV_3CL</a></li> |
|
|
135 |
<li class="toctree-l2"><a class="reference internal" href="data/load_antiviral_drugs.html">load_antiviral_drugs</a></li> |
|
|
136 |
<li class="toctree-l2"><a class="reference internal" href="data/load_broad_repurposing_hub.html">load_broad_repurposing_hub</a></li> |
|
|
137 |
</ul> |
|
|
138 |
</li> |
|
|
139 |
<li class="toctree-l1"><a class="reference internal" href="chemutils.html">DeepPurpose.chemutils</a><ul> |
|
|
140 |
<li class="toctree-l2"><a class="reference internal" href="chem/onek_encoding_unk.html">DeepPurpose.chemutils.onek_encoding_unk</a></li> |
|
|
141 |
<li class="toctree-l2"><a class="reference internal" href="chem/atom_features.html">DeepPurpose.chemutils.atom_features</a></li> |
|
|
142 |
<li class="toctree-l2"><a class="reference internal" href="chem/bond_features.html">DeepPurpose.chemutils.bond_features</a></li> |
|
|
143 |
</ul> |
|
|
144 |
</li> |
|
|
145 |
<li class="toctree-l1"><a class="reference internal" href="oneliner.html">DeepPurpose.oneliner</a><ul> |
|
|
146 |
<li class="toctree-l2"><a class="reference internal" href="oneliner_folder/repurpose.html">DeepPurpose.oneliner.repurpose</a></li> |
|
|
147 |
<li class="toctree-l2"><a class="reference internal" href="oneliner_folder/virtual_screening.html">DeepPurpose.oneliner.virtual_screening</a></li> |
|
|
148 |
</ul> |
|
|
149 |
</li> |
|
|
150 |
<li class="toctree-l1"><a class="reference internal" href="model_helper.html">DeepPurpose.model_helper</a></li> |
|
|
151 |
<li class="toctree-l1"><a class="reference internal" href="utils.html">DeepPurpose.utils</a></li> |
|
|
152 |
</ul> |
|
|
153 |
<p class="caption"><span class="caption-text">Importance Function</span></p> |
|
|
154 |
<ul class="current"> |
|
|
155 |
<li class="toctree-l1"><a class="reference internal" href="model.html">Drug Target Binding Affinity (DTBA) Model</a><ul> |
|
|
156 |
<li class="toctree-l2"><a class="reference internal" href="dtba/classifier.html">Classifier</a></li> |
|
|
157 |
<li class="toctree-l2"><a class="reference internal" href="dtba/dbta.html">Drug Target Binding Affinity (DTBA) Model</a></li> |
|
|
158 |
</ul> |
|
|
159 |
</li> |
|
|
160 |
<li class="toctree-l1"><a class="reference internal" href="encoder.html">Drug/Target Encoder</a><ul> |
|
|
161 |
<li class="toctree-l2"><a class="reference internal" href="encoder.html#drug-encoding">Drug encoding</a></li> |
|
|
162 |
<li class="toctree-l2"><a class="reference internal" href="encoder.html#target-encoding">Target encoding</a></li> |
|
|
163 |
<li class="toctree-l2"><a class="reference internal" href="encoder.html#encoder-model">Encoder Model</a></li> |
|
|
164 |
<li class="toctree-l2"><a class="reference internal" href="encoder.html#technical-details">Technical Details</a><ul> |
|
|
165 |
<li class="toctree-l3"><a class="reference internal" href="encoders/transformer.html">Transformer</a></li> |
|
|
166 |
<li class="toctree-l3"><a class="reference internal" href="encoders/mpnn.html">Message Passing Neural Network (MPNN)</a></li> |
|
|
167 |
<li class="toctree-l3"><a class="reference internal" href="encoders/cnnrnn.html">CNN+RNN</a></li> |
|
|
168 |
<li class="toctree-l3"><a class="reference internal" href="encoders/cnn.html">CNN</a></li> |
|
|
169 |
<li class="toctree-l3"><a class="reference internal" href="encoders/mlp.html">MLP</a></li> |
|
|
170 |
</ul> |
|
|
171 |
</li> |
|
|
172 |
</ul> |
|
|
173 |
</li> |
|
|
174 |
<li class="toctree-l1"><a class="reference internal" href="process_data.html">Processing Data</a><ul> |
|
|
175 |
<li class="toctree-l2"><a class="reference internal" href="process_data.html#drug-target-binding-benchmark-dataset">Drug-Target Binding Benchmark Dataset</a></li> |
|
|
176 |
<li class="toctree-l2"><a class="reference internal" href="process_data.html#repurposing-dataset">Repurposing Dataset</a></li> |
|
|
177 |
<li class="toctree-l2"><a class="reference internal" href="process_data.html#bioassay-data-for-covid-19">Bioassay Data for COVID-19</a></li> |
|
|
178 |
<li class="toctree-l2"><a class="reference internal" href="process_data.html#covid-19-targets">COVID-19 Targets</a><ul> |
|
|
179 |
<li class="toctree-l3"><a class="reference internal" href="data/read_file_training_dataset_bioassay.html">read_file_training_dataset_bioassay</a></li> |
|
|
180 |
<li class="toctree-l3"><a class="reference internal" href="data/read_file_training_dataset_drug_target_pairs.html">read_file_training_dataset_drug_target_pairs</a></li> |
|
|
181 |
<li class="toctree-l3"><a class="reference internal" href="data/read_file_virtual_screening_drug_target_pairs.html">read_file_virtual_screening_drug_target_pairs</a></li> |
|
|
182 |
<li class="toctree-l3"><a class="reference internal" href="data/read_file_repurposing_library.html">load bioarray dataset (read_file_training_dataset_bioassay)</a></li> |
|
|
183 |
<li class="toctree-l3"><a class="reference internal" href="data/read_file_target_sequence.html">read_file_target_sequence</a></li> |
|
|
184 |
<li class="toctree-l3"><a class="reference internal" href="data/download_BindingDB.html">download_DrugTargetCommons</a></li> |
|
|
185 |
<li class="toctree-l3"><a class="reference internal" href="data/process_BindingDB.html">process_BindingDB</a></li> |
|
|
186 |
<li class="toctree-l3"><a class="reference internal" href="data/load_process_DAVIS.html">load_process_DAVIS</a></li> |
|
|
187 |
<li class="toctree-l3"><a class="reference internal" href="data/load_process_KIBA.html">load_process_KIBA</a></li> |
|
|
188 |
<li class="toctree-l3"><a class="reference internal" href="data/load_AID1706_txt_file.html">load_AID1706_txt_file</a></li> |
|
|
189 |
<li class="toctree-l3"><a class="reference internal" href="data/load_AID1706_SARS_CoV_3CL.html">load_AID1706_SARS_CoV_3CL</a></li> |
|
|
190 |
<li class="toctree-l3"><a class="reference internal" href="data/load_antiviral_drugs.html">load_antiviral_drugs</a></li> |
|
|
191 |
<li class="toctree-l3"><a class="reference internal" href="data/load_broad_repurposing_hub.html">load_broad_repurposing_hub</a></li> |
|
|
192 |
</ul> |
|
|
193 |
</li> |
|
|
194 |
</ul> |
|
|
195 |
</li> |
|
|
196 |
<li class="toctree-l1 current"><a class="current reference internal" href="#">Configuration</a></li> |
|
|
197 |
<li class="toctree-l1"><a class="reference internal" href="utility_function.html">Utility Function</a></li> |
|
|
198 |
</ul> |
|
|
199 |
|
|
|
200 |
|
|
|
201 |
|
|
|
202 |
</div> |
|
|
203 |
</div> |
|
|
204 |
</nav> |
|
|
205 |
|
|
|
206 |
<section data-toggle="wy-nav-shift" class="wy-nav-content-wrap"> |
|
|
207 |
|
|
|
208 |
|
|
|
209 |
<nav class="wy-nav-top" aria-label="top navigation"> |
|
|
210 |
|
|
|
211 |
<i data-toggle="wy-nav-top" class="fa fa-bars"></i> |
|
|
212 |
<a href="../index.html">DeepPurpose</a> |
|
|
213 |
|
|
|
214 |
</nav> |
|
|
215 |
|
|
|
216 |
|
|
|
217 |
<div class="wy-nav-content"> |
|
|
218 |
|
|
|
219 |
<div class="rst-content"> |
|
|
220 |
|
|
|
221 |
|
|
|
222 |
|
|
|
223 |
|
|
|
224 |
|
|
|
225 |
|
|
|
226 |
|
|
|
227 |
|
|
|
228 |
|
|
|
229 |
|
|
|
230 |
|
|
|
231 |
|
|
|
232 |
|
|
|
233 |
|
|
|
234 |
|
|
|
235 |
|
|
|
236 |
|
|
|
237 |
<div role="navigation" aria-label="breadcrumbs navigation"> |
|
|
238 |
|
|
|
239 |
<ul class="wy-breadcrumbs"> |
|
|
240 |
|
|
|
241 |
<li><a href="../index.html">Docs</a> »</li> |
|
|
242 |
|
|
|
243 |
<li>Configuration</li> |
|
|
244 |
|
|
|
245 |
|
|
|
246 |
<li class="wy-breadcrumbs-aside"> |
|
|
247 |
|
|
|
248 |
|
|
|
249 |
<a href="../_sources/notes/configuration.rst.txt" rel="nofollow"> View page source</a> |
|
|
250 |
|
|
|
251 |
|
|
|
252 |
</li> |
|
|
253 |
|
|
|
254 |
</ul> |
|
|
255 |
|
|
|
256 |
|
|
|
257 |
<hr/> |
|
|
258 |
</div> |
|
|
259 |
<div role="main" class="document" itemscope="itemscope" itemtype="http://schema.org/Article"> |
|
|
260 |
<div itemprop="articleBody"> |
|
|
261 |
|
|
|
262 |
<div class="section" id="configuration"> |
|
|
263 |
<h1>Configuration<a class="headerlink" href="#configuration" title="Permalink to this headline">¶</a></h1> |
|
|
264 |
<p><strong>generate_config</strong> generate all the configuration that can be used in learning and inference.</p> |
|
|
265 |
<div class="highlight-python notranslate"><div class="highlight"><pre><span></span><span class="n">utils</span><span class="o">.</span><span class="n">generate_config</span><span class="p">(</span> |
|
|
266 |
<span class="n">drug_encoding</span><span class="p">,</span> |
|
|
267 |
<span class="n">target_encoding</span><span class="p">,</span> |
|
|
268 |
<span class="n">result_folder</span> <span class="o">=</span> <span class="s2">"./result/"</span><span class="p">,</span> |
|
|
269 |
<span class="n">input_dim_drug</span> <span class="o">=</span> <span class="mi">1024</span><span class="p">,</span> |
|
|
270 |
<span class="n">input_dim_protein</span> <span class="o">=</span> <span class="mi">8420</span><span class="p">,</span> |
|
|
271 |
<span class="n">hidden_dim_drug</span> <span class="o">=</span> <span class="mi">256</span><span class="p">,</span> |
|
|
272 |
<span class="n">hidden_dim_protein</span> <span class="o">=</span> <span class="mi">256</span><span class="p">,</span> |
|
|
273 |
<span class="n">cls_hidden_dims</span> <span class="o">=</span> <span class="p">[</span><span class="mi">1024</span><span class="p">,</span> <span class="mi">1024</span><span class="p">,</span> <span class="mi">512</span><span class="p">],</span> |
|
|
274 |
<span class="n">mlp_hidden_dims_drug</span> <span class="o">=</span> <span class="p">[</span><span class="mi">1024</span><span class="p">,</span> <span class="mi">256</span><span class="p">,</span> <span class="mi">64</span><span class="p">],</span> |
|
|
275 |
<span class="n">mlp_hidden_dims_target</span> <span class="o">=</span> <span class="p">[</span><span class="mi">1024</span><span class="p">,</span> <span class="mi">256</span><span class="p">,</span> <span class="mi">64</span><span class="p">],</span> |
|
|
276 |
<span class="n">batch_size</span> <span class="o">=</span> <span class="mi">256</span><span class="p">,</span> |
|
|
277 |
<span class="n">train_epoch</span> <span class="o">=</span> <span class="mi">10</span><span class="p">,</span> |
|
|
278 |
<span class="n">test_every_X_epoch</span> <span class="o">=</span> <span class="mi">20</span><span class="p">,</span> |
|
|
279 |
<span class="n">LR</span> <span class="o">=</span> <span class="mf">1e-4</span><span class="p">,</span> |
|
|
280 |
<span class="n">transformer_emb_size_drug</span> <span class="o">=</span> <span class="mi">128</span><span class="p">,</span> |
|
|
281 |
<span class="n">transformer_intermediate_size_drug</span> <span class="o">=</span> <span class="mi">512</span><span class="p">,</span> |
|
|
282 |
<span class="n">transformer_num_attention_heads_drug</span> <span class="o">=</span> <span class="mi">8</span><span class="p">,</span> |
|
|
283 |
<span class="n">transformer_n_layer_drug</span> <span class="o">=</span> <span class="mi">8</span><span class="p">,</span> |
|
|
284 |
<span class="n">transformer_emb_size_target</span> <span class="o">=</span> <span class="mi">128</span><span class="p">,</span> |
|
|
285 |
<span class="n">transformer_intermediate_size_target</span> <span class="o">=</span> <span class="mi">512</span><span class="p">,</span> |
|
|
286 |
<span class="n">transformer_num_attention_heads_target</span> <span class="o">=</span> <span class="mi">8</span><span class="p">,</span> |
|
|
287 |
<span class="n">transformer_n_layer_target</span> <span class="o">=</span> <span class="mi">4</span><span class="p">,</span> |
|
|
288 |
<span class="n">transformer_dropout_rate</span> <span class="o">=</span> <span class="mf">0.1</span><span class="p">,</span> |
|
|
289 |
<span class="n">transformer_attention_probs_dropout</span> <span class="o">=</span> <span class="mf">0.1</span><span class="p">,</span> |
|
|
290 |
<span class="n">transformer_hidden_dropout_rate</span> <span class="o">=</span> <span class="mf">0.1</span><span class="p">,</span> |
|
|
291 |
<span class="n">mpnn_hidden_size</span> <span class="o">=</span> <span class="mi">50</span><span class="p">,</span> |
|
|
292 |
<span class="n">mpnn_depth</span> <span class="o">=</span> <span class="mi">3</span><span class="p">,</span> |
|
|
293 |
<span class="n">cnn_drug_filters</span> <span class="o">=</span> <span class="p">[</span><span class="mi">32</span><span class="p">,</span><span class="mi">64</span><span class="p">,</span><span class="mi">96</span><span class="p">],</span> |
|
|
294 |
<span class="n">cnn_drug_kernels</span> <span class="o">=</span> <span class="p">[</span><span class="mi">4</span><span class="p">,</span><span class="mi">6</span><span class="p">,</span><span class="mi">8</span><span class="p">],</span> |
|
|
295 |
<span class="n">cnn_target_filters</span> <span class="o">=</span> <span class="p">[</span><span class="mi">32</span><span class="p">,</span><span class="mi">64</span><span class="p">,</span><span class="mi">96</span><span class="p">],</span> |
|
|
296 |
<span class="n">cnn_target_kernels</span> <span class="o">=</span> <span class="p">[</span><span class="mi">4</span><span class="p">,</span><span class="mi">8</span><span class="p">,</span><span class="mi">12</span><span class="p">],</span> |
|
|
297 |
<span class="n">rnn_Use_GRU_LSTM_drug</span> <span class="o">=</span> <span class="s1">'GRU'</span><span class="p">,</span> |
|
|
298 |
<span class="n">rnn_drug_hid_dim</span> <span class="o">=</span> <span class="mi">64</span><span class="p">,</span> |
|
|
299 |
<span class="n">rnn_drug_n_layers</span> <span class="o">=</span> <span class="mi">2</span><span class="p">,</span> |
|
|
300 |
<span class="n">rnn_drug_bidirectional</span> <span class="o">=</span> <span class="kc">True</span><span class="p">,</span> |
|
|
301 |
<span class="n">rnn_Use_GRU_LSTM_target</span> <span class="o">=</span> <span class="s1">'GRU'</span><span class="p">,</span> |
|
|
302 |
<span class="n">rnn_target_hid_dim</span> <span class="o">=</span> <span class="mi">64</span><span class="p">,</span> |
|
|
303 |
<span class="n">rnn_target_n_layers</span> <span class="o">=</span> <span class="mi">2</span><span class="p">,</span> |
|
|
304 |
<span class="n">rnn_target_bidirectional</span> <span class="o">=</span> <span class="kc">True</span> |
|
|
305 |
<span class="p">)</span> |
|
|
306 |
</pre></div> |
|
|
307 |
</div> |
|
|
308 |
<ul class="simple"> |
|
|
309 |
<li><p><strong>drug_encoding</strong> (str) - Encoder mode for drug. It can be “transformer”, “MPNN”, “CNN”, “CNN_RNN” …,</p></li> |
|
|
310 |
<li><p><strong>target_encoding</strong> (str) - Encoder mode for protein. It can be “transformer”, “CNN”, “CNN_RNN” …,</p></li> |
|
|
311 |
<li><p><strong>input_dim_drug</strong> (int) - Dimension of input drug feature.</p></li> |
|
|
312 |
<li><p><strong>input_dim_protein</strong> (int) - Dimension of input protein feature.</p></li> |
|
|
313 |
<li><p><strong>hidden_dim_drug</strong> (int) - Dimension of hidden layer of drug feature.</p></li> |
|
|
314 |
<li><p><strong>hidden_dim_protein</strong> (int) - Dimension of hidden layer of protein feature.</p></li> |
|
|
315 |
<li><p><strong>batch_size</strong> (int) - batch size</p></li> |
|
|
316 |
<li><p><strong>train_epoch</strong> (int) - training epoch</p></li> |
|
|
317 |
<li><p><strong>test_every_X_epoch</strong> (int) - test every X epochs</p></li> |
|
|
318 |
<li><p><strong>LR</strong> (float) - Learning rate.</p></li> |
|
|
319 |
<li><p><strong>cls_hidden_dims</strong> (list of int) - hidden dimensions of classifier.</p></li> |
|
|
320 |
<li><p><strong>mlp_hidden_dims_drug</strong> (list of int) - hidden dimension of MLP when encoding drug.</p></li> |
|
|
321 |
<li><p><strong>mlp_hidden_dims_target</strong> (list of int) - hidden dimension of MLP when encoding protein.</p></li> |
|
|
322 |
<li><p><strong>transformer_emb_size_drug</strong> (int) - embedding size of transformer when encoding drug.</p></li> |
|
|
323 |
<li><p><strong>transformer_intermediate_size_drug</strong> (int) -</p></li> |
|
|
324 |
<li><p><strong>transformer_num_attention_heads_drug</strong> (int) -</p></li> |
|
|
325 |
<li><p><strong>transformer_n_layer_drug</strong> (int) -</p></li> |
|
|
326 |
<li><p><strong>transformer_emb_size_target</strong> (int) -</p></li> |
|
|
327 |
<li><p><strong>transformer_intermediate_size_target</strong> (int) -</p></li> |
|
|
328 |
<li><p><strong>transformer_num_attention_heads_target</strong> (int) -</p></li> |
|
|
329 |
<li><p><strong>transformer_n_layer_target</strong> (int) -</p></li> |
|
|
330 |
<li><p><strong>transformer_dropout_rate</strong> (float) -</p></li> |
|
|
331 |
</ul> |
|
|
332 |
</div> |
|
|
333 |
|
|
|
334 |
|
|
|
335 |
</div> |
|
|
336 |
|
|
|
337 |
</div> |
|
|
338 |
<footer> |
|
|
339 |
|
|
|
340 |
<div class="rst-footer-buttons" role="navigation" aria-label="footer navigation"> |
|
|
341 |
|
|
|
342 |
<a href="utility_function.html" class="btn btn-neutral float-right" title="Utility Function" accesskey="n" rel="next">Next <span class="fa fa-arrow-circle-right"></span></a> |
|
|
343 |
|
|
|
344 |
|
|
|
345 |
<a href="process_data.html" class="btn btn-neutral float-left" title="Processing Data" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left"></span> Previous</a> |
|
|
346 |
|
|
|
347 |
</div> |
|
|
348 |
|
|
|
349 |
|
|
|
350 |
<hr/> |
|
|
351 |
|
|
|
352 |
<div role="contentinfo"> |
|
|
353 |
<p> |
|
|
354 |
© Copyright 2020, Kexin Huang, Tianfan Fu |
|
|
355 |
|
|
|
356 |
</p> |
|
|
357 |
</div> |
|
|
358 |
Built with <a href="http://sphinx-doc.org/">Sphinx</a> using a <a href="https://github.com/rtfd/sphinx_rtd_theme">theme</a> provided by <a href="https://readthedocs.org">Read the Docs</a>. |
|
|
359 |
|
|
|
360 |
</footer> |
|
|
361 |
|
|
|
362 |
</div> |
|
|
363 |
</div> |
|
|
364 |
|
|
|
365 |
</section> |
|
|
366 |
|
|
|
367 |
</div> |
|
|
368 |
|
|
|
369 |
|
|
|
370 |
|
|
|
371 |
<script type="text/javascript"> |
|
|
372 |
jQuery(function () { |
|
|
373 |
SphinxRtdTheme.Navigation.enable(true); |
|
|
374 |
}); |
|
|
375 |
</script> |
|
|
376 |
|
|
|
377 |
|
|
|
378 |
|
|
|
379 |
|
|
|
380 |
|
|
|
381 |
|
|
|
382 |
</body> |
|
|
383 |
</html> |