@@ -293,7 +293,7 @@ <h1>Source code for torch_molecule.predictor.lstm.modeling_lstm</h1><div class="
293
293
< span class ="p "> }</ span >
294
294
295
295
< div class ="viewcode-block " id ="LSTMMolecularPredictor ">
296
- < a class ="viewcode-back " href ="../../../../api/predictor.html#torch_molecule.predictor.lstm.LSTMMolecularPredictor "> [docs]</ a >
296
+ < a class ="viewcode-back " href ="../../../../api/predictor.html#torch_molecule.predictor.lstm.modeling_lstm. LSTMMolecularPredictor "> [docs]</ a >
297
297
< span class ="nd "> @dataclass</ span >
298
298
< span class ="k "> class</ span > < span class ="w "> </ span > < span class ="nc "> LSTMMolecularPredictor</ span > < span class ="p "> (</ span > < span class ="n "> BaseMolecularPredictor</ span > < span class ="p "> ):</ span >
299
299
< span class ="w "> </ span > < span class ="sd "> """This predictor implements a LSTM model for molecular property prediction tasks.</ span >
@@ -498,7 +498,7 @@ <h1>Source code for torch_molecule.predictor.lstm.modeling_lstm</h1><div class="
498
498
< span class ="k "> return</ span > < span class ="n "> DEFAULT_LSTM_SEARCH_SPACES</ span >
499
499
500
500
< div class ="viewcode-block " id ="LSTMMolecularPredictor.autofit ">
501
- < a class ="viewcode-back " href ="../../../../api/predictor.html#torch_molecule.predictor.lstm.LSTMMolecularPredictor.autofit "> [docs]</ a >
501
+ < a class ="viewcode-back " href ="../../../../api/predictor.html#torch_molecule.predictor.lstm.modeling_lstm. LSTMMolecularPredictor.autofit "> [docs]</ a >
502
502
< span class ="k "> def</ span > < span class ="w "> </ span > < span class ="nf "> autofit</ span > < span class ="p "> (</ span >
503
503
< span class ="bp "> self</ span > < span class ="p "> ,</ span >
504
504
< span class ="n "> X_train</ span > < span class ="p "> :</ span > < span class ="n "> List</ span > < span class ="p "> [</ span > < span class ="nb "> str</ span > < span class ="p "> ],</ span >
@@ -656,7 +656,7 @@ <h1>Source code for torch_molecule.predictor.lstm.modeling_lstm</h1><div class="
656
656
657
657
658
658
< div class ="viewcode-block " id ="LSTMMolecularPredictor.fit ">
659
- < a class ="viewcode-back " href ="../../../../api/predictor.html#torch_molecule.predictor.lstm.LSTMMolecularPredictor.fit "> [docs]</ a >
659
+ < a class ="viewcode-back " href ="../../../../api/predictor.html#torch_molecule.predictor.lstm.modeling_lstm. LSTMMolecularPredictor.fit "> [docs]</ a >
660
660
< span class ="k "> def</ span > < span class ="w "> </ span > < span class ="nf "> fit</ span > < span class ="p "> (</ span >
661
661
< span class ="bp "> self</ span > < span class ="p "> ,</ span >
662
662
< span class ="n "> X_train</ span > < span class ="p "> :</ span > < span class ="n "> List</ span > < span class ="p "> [</ span > < span class ="nb "> str</ span > < span class ="p "> ],</ span >
@@ -777,7 +777,7 @@ <h1>Source code for torch_molecule.predictor.lstm.modeling_lstm</h1><div class="
777
777
778
778
779
779
< div class ="viewcode-block " id ="LSTMMolecularPredictor.predict ">
780
- < a class ="viewcode-back " href ="../../../../api/predictor.html#torch_molecule.predictor.lstm.LSTMMolecularPredictor.predict "> [docs]</ a >
780
+ < a class ="viewcode-back " href ="../../../../api/predictor.html#torch_molecule.predictor.lstm.modeling_lstm. LSTMMolecularPredictor.predict "> [docs]</ a >
781
781
< 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 "> List</ span > < span class ="p "> [</ span > < span class ="nb "> str</ span > < span class ="p "> ])</ span > < span class ="o "> -></ span > < span class ="n "> Dict</ span > < span class ="p "> [</ span > < span class ="nb "> str</ span > < span class ="p "> ,</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ]:</ span >
782
782
< span class ="w "> </ span > < span class ="sd "> """Make predictions using the fitted model.</ span >
783
783
0 commit comments