chemprop.models.utils

Module Contents

Functions

save_model(path, model)
load_model(path, multicomponent)

chemprop.models.utils.save_model(path, model)

Parameters:

• path (os.PathLike)

• model (chemprop.models.model.MPNN)

Return type:

None

chemprop.models.utils.load_model(path, multicomponent)

Parameters:

• path (os.PathLike)

• multicomponent (bool)

Return type:

chemprop.models.model.MPNN