Chemperium Tc 2D
PyTorch-порт ансамбля для критической температуры
Это ансамбль из десяти моделей, который предсказывает критическую температуру (K) молекулы по её SMILES. Веса взяты из опубликованной модели Chemperium liquid/tc/2d авторов Dobbelaere et al. и без изменений перенесены из Keras в PyTorch safetensors.
➕ Ключевые особенности
- 1⃣ Ансамбль из 10 моделей выдаёт среднее предсказание критической температуры.
- 2⃣ Веса сконвертированы без научных изменений; архитектура и SHA-256 исходников записаны в
config.json. - 3⃣ В комплекте
scaler.jsonс min-max масштабированием выхода в K. - 4⃣ PyTorch-порт воспроизводит 278 предсказаний TensorFlow с расхождением до 1.2e-4 K.
- 5⃣ Молекулы с 75 и более связями, считая явные водороды, не предсказываются.
▶ Пример
from screening_properties import get_properties, Component
props = get_properties(
[Component("my-compound", smiles="CCOC(=O)C")],
properties=["critical_temperature"],
required=[]
)
Подробнее
Проверка: 278 предсказаний совпадают с TensorFlow в пределах 1.2e-4 K.Неопределённость — стандартное отклонение ансамбля, это не калиброванный доверительный интервал.
Код оригинала под MIT, веса и данные — CC BY 4.0.
Модель используется в
screening-properties с проверкой SHA-256 файлов.







Комментарии