AttrMaskMolecularEncoder Model

Model Description

• Model Type: AttrMaskMolecularEncoder
• Framework: torch_molecule
• Last Updated: 2025-04-05

Task Summary

Task	Version	Last Updated	Parameters	Metrics
default	0.0.9	2025-04-05	3,593,223

Usage

from torch_molecule import AttrMaskMolecularEncoder

# Load model for specific task
model = AttrMaskMolecularEncoder()
model.load(
    "local_model_dir/AttrMaskMolecularEncoder.pt",
    repo="Einae/AttrMask"
)

# For predictor: Make predictions
# predictions = model.predict(smiles_list)
# For generator: Make generations
# generations = model.generate(n_samples)
# For encoder: Make encodings
# encodings = model.encode(smiles_list)

Tasks Details

default Task

• Current Version: 0.0.9
• Last Updated: 2025-04-05
• Parameters: 3,593,223
• Configuration:

{
  "mask_num": 0,
  "mask_rate": 0.15,
  "encoder_type": "gin-virtual",
  "readout": "sum",
  "num_layer": 5,
  "hidden_size": 300,
  "drop_ratio": 0.5,
  "norm_layer": "batch_norm",
  "batch_size": 128,
  "epochs": 100,
  "learning_rate": 0.001,
  "weight_decay": 0.0,
  "grad_clip_value": null,
  "use_lr_scheduler": false,
  "scheduler_factor": 0.5,
  "scheduler_patience": 5,
  "fitting_epoch": 99,
  "device": {
    "_type": "unknown",
    "repr": "cuda:0"
  },
  "verbose": false,
  "model_name": "AttrMaskMolecularEncoder"
}