graphchem.preprocessing.load_encoder

Loads a pre-saved MoleculeEncoder object from a file.

Parameters

filename : str The path to the saved encoder file.

Returns

MoleculeEncoder The loaded MoleculeEncoder object.

Source code in graphchem/preprocessing/features.py

def load_encoder(filename: str) -> MoleculeEncoder:
    """
    Loads a pre-saved `MoleculeEncoder` object from a file.

    Parameters
    ----------
    filename : str
        The path to the saved encoder file.

    Returns
    -------
    MoleculeEncoder
        The loaded `MoleculeEncoder` object.
    """
    with open(filename, "rb") as inp:
        encoder = pickle.load(inp)
    return encoder