openfl.interface.model.get_model

openfl.interface.model.get_model(plan_config: str, cols_config: str, data_config: str, model_protobuf_path: str)

Initialize TaskRunner and load it with provided model.pbuf.

Contrary to its name, this function returns a TaskRunner instance. The reason for this behavior is the flexibility of the TaskRunner interface and the diversity of the ways we store models in our template workspaces.

Parameters:
  • plan_config (str) – Federated learning plan.

  • cols_config (str) – Authorized collaborator list.

  • data_config (str) – The data set/shard configuration file.

  • model_protobuf_path (str) – The model protobuf to convert.

Returns:

task_runner (instance) – TaskRunner instance.