lmflow.models.text_regression_model

A model maps “text_only” data to float.

Module Contents

Classes

TextRegressionModel

Initializes a TextRegressionModel instance.

class lmflow.models.text_regression_model.TextRegressionModel(model_args, *args, **kwargs)

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.

register_inference_function(inference_func)

Registers a regression function.

inference(inputs: lmflow.datasets.dataset.Dataset)

Gets regression results of a given dataset.

Inputs:

Dataset object, only accept type “text_only”.