Views
No views yet
| Task | Version | Last Updated | Parameters | Metrics |
|---|---|---|---|---|
| default | 0.0.10 | 2025-08-07 | 3,832,927 |
1from torch_molecule import MoamaMolecularEncoder
2
3# Load model for specific task
4model = MoamaMolecularEncoder()
5model.load(
6 "local_model_dir/MoamaMolecularEncoder.pt",
7 repo="Do1Yun/TorchMoleculeEncoderRepo"
8)
9
10# For predictor: Make predictions
11# predictions = model.predict(smiles_list)
12# For generator: Make generations
13# generations = model.generate(n_samples)
14# For encoder: Make encodings
15# encodings = model.encode(smiles_list)1{
2 "mask_rate": 0.15,
3 "lw_rec": 0.5,
4 "encoder_type": "gin-virtual",
5 "readout": "sum",
6 "num_layer": 5,
7 "hidden_size": 300,
8 "drop_ratio": 0.5,
9 "norm_layer": "batch_norm",
10 "batch_size": 32,
11 "epochs": 10,
12 "learning_rate": 0.001,
13 "weight_decay": 0.0,
14 "grad_clip_value": null,
15 "use_lr_scheduler": false,
16 "scheduler_factor": 0.5,
17 "scheduler_patience": 5,
18 "fitting_epoch": 9,
19 "device": {
20 "_type": "unknown",
21 "repr": "cuda:0"
22 },
23 "verbose": false,
24 "model_name": "MoamaMolecularEncoder"
25}