rii dZddlZddlmZddlmZmZmZddlm Z m Z m Z m Z ddl mZGdd eZe dddd d d d ededeedeedeedededeeeeffdZdS)z4XGBoost Experimental Federated Learning related API.N)Thread)AnyDictOptional)_LIB _check_call_deprecate_positional_args make_jcargs) RabitTrackerczeZdZdZeddddddedededeed eed eed ed dfd Z dS)FederatedTrackeraTracker for federated training. Parameters ---------- n_workers : The number of federated workers. port : The port to listen on. secure : Whether this is a secure instance. If True, then the following arguments for SSL must be provided. server_key_path : Path to the server private key file. server_cert_path : Path to the server certificate file. client_cert_path : Path to the client certificate file. N,)server_key_pathserver_cert_pathclient_cert_pathtimeout n_workersportsecurerrrrreturnc tj}t||d||||t|} t t j| tj|||_dS)N federated)rrdmlc_communicatorfederated_securerrrr) ctypesc_void_pr intr rXGTrackerCreatebyrefhandle) selfrrrrrrrr!argss W/var/www/html/bet.cuttalo.com/ml/venv/lib/python3.11/site-packages/xgboost/federated.py__init__zFederatedTracker.__init__%su"")#+--LL     D(v|F/C/CDDEEE ) __name__ __module__ __qualname____doc__r rboolrstrr%r&r$rr s2 *.*.*.   "# #3-#3-  r&rTr)rrrblockingrrrrrrr.rrc xd|i}td|||fD}t|||||||} | |r| dSt | j} d| _| || |S)a|See :py:class:`~xgboost.federated.FederatedTracker` for more info. Parameters ---------- blocking : Block the server until the training is finished. If set to False, the function launches an additional thread and returns the worker arguments. The default is True and a higher level framework is responsible for setting worker parameters. rc3K|]}|duV dS)Nr-).0paths r$ z'run_federated_server..Vs;  Dr&)rrrrrrrN)targetT)allrstartwait_forrdaemonupdate worker_args) rrrrrr.rr#rtrackerthreads r$run_federated_serverr=@s*(3D $&68HIF '))G MMOOOt 7+ , , ,FFM LLNNNKK##%%&&& Kr&)r*r threadingrtypingrrrcorerr r r r;r rrr,r+r=r-r&r$rAsP:: &&&&&&&&&&LLLLLLLLLLLL!!!!!!22222|222j &*&*&*,,,, ,c] , sm , sm ,,,d38n,,,,,,r&