kiwi.lib.pretrain

Module Contents

Classes

Configuration

Base class for all pydantic configs. Used to configure base behaviour of configs.

Functions

pretrain_from_file(filename) → TrainRunInfo	Load options from a config file and call the pretraining procedure.
pretrain_from_configuration(configuration_dict) → TrainRunInfo	Run the entire training pipeline using the configuration options received.

kiwi.lib.pretrain.logger

class kiwi.lib.pretrain.Configuration

Bases: kiwi.lib.train.Configuration

Base class for all pydantic configs. Used to configure base behaviour of configs.

system :TLMSystem.Config

kiwi.lib.pretrain.pretrain_from_file(filename) → TrainRunInfo

Load options from a config file and call the pretraining procedure.

Parameters

filename – of the configuration file.

Returns

object with training information.

kiwi.lib.pretrain.pretrain_from_configuration(configuration_dict) → TrainRunInfo

Run the entire training pipeline using the configuration options received.

Parameters

configuration_dict – dictionary with config options.

Returns

object with training information.

kiwi.lib.predict kiwi.lib.search