golf_federated.server.process.config.model package

golf_federated.server.process.config.model.base module

class golf_federated.server.process.config.model.base.BaseModel(module: object, test_data: numpy.ndarray, test_label: numpy.ndarray, process_unit: str)

Bases: object

Model object class, the class function supports the main operation of model on Server.

get_weight() → List

Get model weight.

Returns:

List: Model weight.

model_aggre(aggregation: golf_federated.server.process.strategy.aggregation.base.BaseFed, parameter: queue.Queue, record: List) → None

Global model aggregation.

Args:

aggregation (golf_federated.server.process.strategy.aggregation.base.BaseFed): Aggregation strategy object. parameter (queue.Queue): Uploaded parameters. record (List): Records of evaluation.

model_eval(evaluation: golf_federated.server.process.strategy.evaluation.base.BaseEval) → None

Global model evaluation.

Args:

evaluation (golf_federated.server.process.strategy.evaluation.base.BaseEval): Evaluation strategy object.

abstract predict() → numpy.ndarray

Abstract method for model prediction.

Returns:

Numpy.ndarray: Prediction result.

update_weight(new_weight: List) → None

Update model weight.

Args:

new_weight (list): Model weight for update.

golf_federated.server.process.config.model.tfmodel module

golf_federated.server.process.config.model.torchmodel module