finetuner.model module#

finetuner.model.get_header()[source]#

Get table header.

Return type:

Tuple[str, …]

finetuner.model.get_row(model_stub)[source]#

Get table row.

Return type:

Tuple[str, …]

finetuner.model.list_model_classes()[source]#
Return type:

Dict[str, ~ModelStubType]

class finetuner.model.SynthesisModels(relation_miner, cross_encoder)[source]#

Bases: object

Class specifying the models to be used in a data synthesis job. :param: relation_miner: The name of the model or list of models to use for

relation mining.

Parameters:

cross_encoder (str) – The name of the model to use as the cross encoder

relation_miner: str | List[str]#
cross_encoder: str#