mhnreact/inference.py

# -*- coding: utf-8 -*-
"""
Author: Philipp Seidl
        ELLIS Unit Linz, LIT AI Lab, Institute for Machine Learning
        Johannes Kepler University Linz
Contact: seidl@ml.jku.at

File contains functions that help prepare and download USPTO-related datasets
"""

# Cell
from .model import ModelConfig, MHN
import torch