toModel method

EncoderModel toModel()

Serializes the encoder to an EncoderModel object.

Implementation

EncoderModel toModel() {
  _checkFitted();

  final map = {
    for (int i = 0; i < _categories.length; i++) _categories[i]: i
  };

  return EncoderModel(
    type: 'onehot',
    labelToIndex: map,
  );
}