deepmol.encoders package

Submodules

deepmol.encoders.label_encoder module

class LabelEncoder

Bases: Transformer

Class that encodes labels as integers.

This class is used to encode labels as integers. This is useful for classification tasks.

encoder

Scikit-learn label encoder.

Type:

SKLabelEncoder

deepmol.encoders.label_one_hot_encoder module

class LabelOneHotEncoder

Bases: Transformer

Class that encodes labels as one-hot vectors. This class is used to encode labels as one-hot vectors. This is useful for classification tasks.

encoder

Scikit-learn one-hot encoder.

Type:

OneHotEncoder

Module contents