nvflare.app_opt.xgboost.histogram_based_v2.defs module

class Constant

Bases: object

CONFIG_TASK_NAME = 'config'

CONFIG_TASK_TIMEOUT = 60

CONF_KEY_CLIENT_RANKS = 'client_ranks'

CONF_KEY_DATA_SPLIT_MODE = 'data_split_mode'

CONF_KEY_DISABLE_VERSION_CHECK = 'xgb_disable_version_check'

CONF_KEY_NUM_ROUNDS = 'num_rounds'

CONF_KEY_SECURE_TRAINING = 'secure_training'

CONF_KEY_WORLD_SIZE = 'world_size'

CONF_KEY_XGB_OPTIONS = 'xgb_options'

CONF_KEY_XGB_PARAMS = 'xgb_params'

DUMMY_BUFFER_SIZE = 4

ERR_INVALID_RANK = -2

ERR_NO_CLIENT_FOR_RANK = -3

ERR_OP_MISMATCH = -1

ERR_TARGET_ERROR = -4

EVENT_AFTER_ALL_GATHER_V = 'xgb.after_all_gather_v'

EVENT_AFTER_BROADCAST = 'xgb.after_broadcast'

EVENT_BEFORE_ALL_GATHER_V = 'xgb.before_all_gather_v'

EVENT_BEFORE_BROADCAST = 'xgb.before_broadcast'

EVENT_XGB_ABORTED = 'xgb.aborted'

EVENT_XGB_JOB_CONFIGURED = 'xgb.job_configured'

EXIT_CODE_CANT_START = 101

EXIT_CODE_JOB_ABORT = 102

HEADER_KEY_ENCRYPTED_DATA = 'xgb.encrypted_data'

HEADER_KEY_HORIZONTAL = 'xgb.horizontal'

HEADER_KEY_IN_AGGR = 'xgb.in_aggr'

HEADER_KEY_ORIGINAL_BUF_SIZE = 'xgb.original_buf_size'

HEADER_KEY_SIZE_DICT = 'xgb.size_dict'

HEADER_KEY_WORLD_SIZE = 'xgb.world_size'

JOB_STATUS_CHECK_INTERVAL = 2.0

MAX_CLIENT_OP_INTERVAL = 600.0

MSG_KEY_EXIT_CODE = 'exit_code'

MSG_KEY_XGB_OP = 'xgb.op'

MSG_KEY_XGB_REQ_ID = 'xgb.req_id'

MSG_KEY_XGB_REQ_RECEIVED = 'xgb.req_received'

MSG_KEY_XGB_REQ_TRY_NUM = 'xgb.req_try_num'

OPCODE_ALL_GATHER = 1

OPCODE_ALL_GATHER_V = 2

OPCODE_ALL_REDUCE = 3

OPCODE_BROADCAST = 4

OPCODE_DONE = 99

OPCODE_NONE = 0

OP_ALL_GATHER = 'all_gather'

OP_ALL_GATHER_V = 'all_gather_v'

OP_ALL_REDUCE = 'all_reduce'

OP_BROADCAST = 'broadcast'

PARAM_KEY_CONFIG_ERROR = 'xgb.config_error'

PARAM_KEY_DATA_SPLIT_MODE = 'xgb.data_split_mode'

PARAM_KEY_DATA_TYPE = 'xgb.data_type'

PARAM_KEY_DISABLE_VERSION_CHECK = 'xgb.disable_version_check'

PARAM_KEY_EVENT = 'xgb.event'

PARAM_KEY_HEADERS = 'xgb.headers'

PARAM_KEY_RANK = 'xgb.rank'

PARAM_KEY_RCV_BUF = 'xgb.rcv_buf'

PARAM_KEY_REDUCE_OP = 'xgb.reduce_op'

PARAM_KEY_REPLY = 'xgb.reply'

PARAM_KEY_REQUEST = 'xgb.request'

PARAM_KEY_ROOT = 'xgb.root'

PARAM_KEY_SECURE_TRAINING = 'xgb.secure_training'

PARAM_KEY_SEND_BUF = 'xgb.send_buf'

PARAM_KEY_SEQ = 'xgb.seq'

RUNNER_CTX_CLIENT_NAME = 'client_name'

RUNNER_CTX_DATA_SPLIT_MODE = 'data_split_mode'

RUNNER_CTX_MODEL_DIR = 'model_dir'

RUNNER_CTX_NUM_ROUNDS = 'num_rounds'

RUNNER_CTX_PORT = 'port'

RUNNER_CTX_RANK = 'rank'

RUNNER_CTX_SECURE_TRAINING = 'secure_training'

RUNNER_CTX_SERVER_ADDR = 'server_addr'

RUNNER_CTX_WORLD_SIZE = 'world_size'

RUNNER_CTX_XGB_DISABLE_VERSION_CHECK = 'xgb_disable_version_check'

RUNNER_CTX_XGB_OPTIONS = 'xgb_options'

RUNNER_CTX_XGB_PARAMS = 'xgb_params'

START_TASK_NAME = 'start'

START_TASK_TIMEOUT = 10

TASK_CHECK_INTERVAL = 0.5

TOPIC_CLIENT_DONE = 'xgb.client_done'

TOPIC_XGB_REQUEST = 'xgb.request'

TOPIC_XGB_REQUEST_CHECK = 'xgb.req_check'

WORKFLOW_PROGRESS_TIMEOUT = 3600.0

XGB_SERVER_READY_TIMEOUT = 10.0