golf_federated.client.process.config.trainer package

golf_federated.client.process.config.trainer.base module

class golf_federated.client.process.config.trainer.base.BaseTrainer(mode: str)

Bases: object

Trainer object class, the class function supports the main operation of trainer on Client.

abstract get_model() → List

Abstract method for model weight getting.

Returns:

List: Model weight.

abstract get_train_data() → numpy.ndarray

Abstract method for data values getting.

Returns:

Numpy.ndarray: Data values.

abstract get_train_label() → numpy.ndarray

Abstract method for data labels getting.

Returns:

Numpy.ndarray: Data labels.

abstract predict(data: numpy.ndarray) → numpy.ndarray

Abstract method for prediction.

Args:

data (numpy.ndarray): Data values for prediction.

Returns:

Numpy.ndarray: Prediction result.

abstract stop_trainer() → None

Abstract method for trainer stopping.

abstract train() → None

Abstract method for training.

abstract update_model(new_weight: List)

Abstract method for model weight update.

Args:

new_weight (List): Model weight for update.

golf_federated.client.process.config.trainer.direct module

golf_federated.client.process.config.trainer.docker module