205
205
< p class ="caption " role ="heading "> < span class ="caption-text "> API Documentation:</ span > </ p >
206
206
< ul >
207
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 >
208
+ < li class ="toctree-l2 "> < a class ="reference internal " href ="../../../api/predictor.html "> Molecular Property Prediction Models</ a > </ li >
209
+ < li class ="toctree-l2 "> < a class ="reference internal " href ="../../../api/generator.html "> Molecular Generation Models</ a > </ li >
210
+ < li class ="toctree-l2 "> < a class ="reference internal " href ="../../../api/encoder.html "> Molecular Representation Learning / Encoding Models</ a > </ li >
212
211
< 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 >
212
+ < li class ="toctree-l2 "> < a class ="reference internal " href ="../../../api/utils.html "> Utility Functions</ a > </ li >
213
+ < li class ="toctree-l2 "> < a class ="reference internal " href ="../../../api/base.html "> Base Modules</ a > </ li >
214
214
</ ul >
215
215
</ li >
216
216
</ ul >
@@ -269,7 +269,7 @@ <h1>Source code for torch_molecule.base.predictor</h1><div class="highlight"><pr
269
269
< span class ="kn "> from</ span > < span class ="w "> </ span > < span class ="nn "> ..base.base</ span > < span class ="w "> </ span > < span class ="kn "> import</ span > < span class ="n "> BaseModel</ span >
270
270
271
271
< div class ="viewcode-block " id ="BaseMolecularPredictor ">
272
- < a class ="viewcode-back " href ="../../../api/utils .html#torch_molecule.base.BaseMolecularPredictor "> [docs]</ a >
272
+ < a class ="viewcode-back " href ="../../../api/base .html#torch_molecule.base.BaseMolecularPredictor "> [docs]</ a >
273
273
< span class ="nd "> @dataclass</ span >
274
274
< span class ="k "> class</ span > < span class ="w "> </ span > < span class ="nc "> BaseMolecularPredictor</ span > < span class ="p "> (</ span > < span class ="n "> BaseModel</ span > < span class ="p "> ,</ span > < span class ="n "> ABC</ span > < span class ="p "> ):</ span >
275
275
< span class ="w "> </ span > < span class ="sd "> """Base class for molecular discovery estimators."""</ span >
@@ -294,21 +294,21 @@ <h1>Source code for torch_molecule.base.predictor</h1><div class="highlight"><pr
294
294
< span class ="k "> return</ span > < span class ="nb "> super</ span > < span class ="p "> ()</ span > < span class ="o "> .</ span > < span class ="n "> _get_param_names</ span > < span class ="p "> ()</ span > < span class ="o "> +</ span > < span class ="p "> [</ span > < span class ="s2 "> "num_task"</ span > < span class ="p "> ,</ span > < span class ="s2 "> "task_type"</ span > < span class ="p "> ]</ span >
295
295
296
296
< div class ="viewcode-block " id ="BaseMolecularPredictor.autofit ">
297
- < a class ="viewcode-back " href ="../../../api/utils .html#torch_molecule.base.BaseMolecularPredictor.autofit "> [docs]</ a >
297
+ < a class ="viewcode-back " href ="../../../api/base .html#torch_molecule.base.BaseMolecularPredictor.autofit "> [docs]</ a >
298
298
< span class ="nd "> @abstractmethod</ span >
299
299
< span class ="k "> def</ span > < span class ="w "> </ span > < span class ="nf "> autofit</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="p "> ,</ span > < span class ="n "> X_train</ span > < span class ="p "> ,</ span > < span class ="n "> y_train</ span > < span class ="p "> ,</ span > < span class ="n "> X_val</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span > < span class ="p "> ,</ span > < span class ="n "> y_val</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span > < span class ="p "> ,</ span > < span class ="n "> search_parameters</ span > < span class ="p "> :</ span > < span class ="n "> Optional</ span > < span class ="p "> [</ span > < span class ="nb "> dict</ span > < span class ="p "> ]</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span > < span class ="p "> ,</ span > < span class ="n "> n_trials</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="o "> =</ span > < span class ="mi "> 10</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="s2 "> "BaseMolecularPredictor"</ span > < span class ="p "> :</ span >
300
300
< span class ="k "> pass</ span > </ div >
301
301
302
302
303
303
< div class ="viewcode-block " id ="BaseMolecularPredictor.fit ">
304
- < a class ="viewcode-back " href ="../../../api/utils .html#torch_molecule.base.BaseMolecularPredictor.fit "> [docs]</ a >
304
+ < a class ="viewcode-back " href ="../../../api/base .html#torch_molecule.base.BaseMolecularPredictor.fit "> [docs]</ a >
305
305
< span class ="nd "> @abstractmethod</ span >
306
306
< 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_train</ span > < span class ="p "> ,</ span > < span class ="n "> y_train</ span > < span class ="p "> ,</ span > < span class ="n "> X_val</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span > < span class ="p "> ,</ span > < span class ="n "> y_val</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span > < span class ="p "> ,</ span > < span class ="n "> search_parameters</ span > < span class ="p "> :</ span > < span class ="n "> Optional</ span > < span class ="p "> [</ span > < span class ="nb "> dict</ span > < span class ="p "> ]</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span > < span class ="p "> ,</ span > < span class ="n "> n_trials</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="o "> =</ span > < span class ="mi "> 10</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="s2 "> "BaseMolecularPredictor"</ span > < span class ="p "> :</ span >
307
307
< span class ="k "> pass</ span > </ div >
308
308
309
309
310
310
< div class ="viewcode-block " id ="BaseMolecularPredictor.predict ">
311
- < a class ="viewcode-back " href ="../../../api/utils .html#torch_molecule.base.BaseMolecularPredictor.predict "> [docs]</ a >
311
+ < a class ="viewcode-back " href ="../../../api/base .html#torch_molecule.base.BaseMolecularPredictor.predict "> [docs]</ a >
312
312
< span class ="nd "> @abstractmethod</ span >
313
313
< span class ="k "> def</ span > < span class ="w "> </ span > < span class ="nf "> predict</ 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 "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> :</ span >
314
314
< span class ="k "> pass</ span > </ div >
0 commit comments