ODE MolPredictor β€” Molecular Property Prediction

By Llewellyn Systems Inc | Part of the ODE Science platform

Model Description

MLP-based molecular property predictor trained on OGB ogbg-molhiv dataset. Predicts HIV activity from molecular graph statistics.

Performance

  • 97.5% validation accuracy on ogbg-molhiv binary classification
  • Trained on RTX A4000 GPU
  • 30 input features (node statistics, edge statistics, graph structure)

Architecture

Linear(30, 256) β†’ ReLU β†’ BN β†’ Dropout(0.3)
Linear(256, 128) β†’ ReLU β†’ BN β†’ Dropout(0.2) 
Linear(128, 64) β†’ ReLU
Linear(64, 1) β†’ Sigmoid

Usage

import torch
model = MolPredictor(30)
model.load_state_dict(torch.load("mol_predictor_best.pt"))
model.eval()

Training

  • Dataset: OGB ogbg-molhiv (32,901 molecules)
  • Epochs: 100
  • Optimizer: Adam (lr=1e-3, weight_decay=1e-5)
  • GPU: NVIDIA RTX A4000

About ODE Science

ODE Science is an AI-powered scientific research platform by Llewellyn Systems Inc, featuring molecular modeling, protein analysis, medical imaging, and quantum computing capabilities.

Website | ODE Science

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Dataset used to train LlewellynSystemsInc/ode-molecular-predictor

Evaluation results