DashAI.back.converters.hugging_face_wrapper.HuggingFaceWrapper
- class HuggingFaceWrapper(**kwargs)[source]
Abstract base wrapper for HuggingFace transformers.
Methods
__init__
(**kwargs)fit
(x[, y])Validate parameters and prepare for transformation.
get_schema
()Generates the component related Json Schema.
transform
(x[, y])Transform the input data using the model.
validate_and_transform
(raw_data)It takes the data given by the user to initialize the model and returns it with all the objects that the model needs to work.
Attributes
TYPE