chemprop.featurizers.base

Module Contents

Classes

Featurizer	An Featurizer featurizes inputs type S into outputs of
VectorFeaturizer	An Featurizer featurizes inputs type S into outputs of
GraphFeaturizer	An Featurizer featurizes inputs type S into outputs of

Attributes

S
T

chemprop.featurizers.base.S

chemprop.featurizers.base.T

class chemprop.featurizers.base.Featurizer

Bases: Generic[S, T]

An Featurizer featurizes inputs type S into outputs of type T.

abstract __call__(input, *args, **kwargs)

featurize an input

Parameters:

input (S)

Return type:

T

class chemprop.featurizers.base.VectorFeaturizer

Bases: Featurizer[S, numpy.ndarray], collections.abc.Sized

An Featurizer featurizes inputs type S into outputs of type T.

class chemprop.featurizers.base.GraphFeaturizer

Bases: Featurizer[S, chemprop.data.molgraph.MolGraph]

An Featurizer featurizes inputs type S into outputs of type T.

abstract property shape: tuple[int, int]

Return type:

tuple[int, int]