converters.sklearn._one_hot_encoder

Module Contents

Functions

convert(model,input_features,output_features) Convert a one-hot-encoder model to the protobuf spec.
update_dimension(model,input_dimension) Given a model that takes an array of dimension input_dimension, returns
get_input_dimension(model)
convert(model, input_features, output_features)

Convert a one-hot-encoder model to the protobuf spec.

model: OneHotEncoder
A trained one-hot encoder model.
input_features: str, optional
Name of the input column.
output_features: str, optional
Name of the output column.
model_spec: An object of type Model_pb.
Protobuf representation of the model
update_dimension(model, input_dimension)

Given a model that takes an array of dimension input_dimension, returns the output dimension.

get_input_dimension(model)