lmflow.models.text_regression_model#
A model maps “text_only” data to float.
Classes#
Initializes a TextRegressionModel instance. |
Module Contents#
- class lmflow.models.text_regression_model.TextRegressionModel(model_args, *args, **kwargs)[source]#
Bases:
lmflow.models.regression_model.RegressionModel
Initializes a TextRegressionModel instance.
- Parameters:
- model_args
Model arguments such as model name, path, revision, etc.
- argsOptional.
Positional arguments.
- kwargsOptional.
Keyword arguments.
- inference(inputs: lmflow.datasets.dataset.Dataset)[source]#
Gets regression results of a given dataset.
- Inputs:
Dataset object, only accept type “text_only”.