
Test_dataset ( np.ndarray) – Testing data, a tuple given in the form (x_test, test) or a datagenerator of of type, List evaluate ( self, test_dataset, **kwargs ) ¶Įvaluates the model given testing data Parameters Node ( dict) – A tree root to start searching for leaf node X_test ( category) – A data instance in the format as expected by the model

ModelUpdate predict ( self, x_test, node=None ) ¶ Generate a ModelUpdate object that will be sent to the entities ReturnsĬounts - a list of counts for the latest query Return type New_list_of_features ( list) – Optional, new list of feature under consideration New_feature_values ( list) – Optional, new range of feature values
TORCH NN SEQUENTIAL POINTER UPDATE
Model_update ( ModelUpdate) – Optional, an ModelUpdate object that contains information to update the DTFLModel. Their corresponding range of feature values. None update_model ( model_update=None, new_feature_values=None, new_list_of_features=None ) ¶ Train_data ( ) – Training datasetįit_params ( dict) – A dictionary containing the aggregator query information, e.g., ‘splits’, which is a list of node split candidates in the current tree ‘list_of_labels’, which is a list of labels needs count information None fit_model ( train_data, fit_params, **kwargs ) ¶Ĭomputes the count according to split information Kwargs ( dict) – A dictionary contains other parameter settings on to initialize a decision tree model. Latest_counts ( list) – A list of last_counts saved from last query call, served as party model_update information

Model_spec ( dict) – Specification of the tree model Model_name ( str) – String specifying the name of the model

If an initial tree structure is given, model_spec is optional. For the given model_spec, it will extract the data information required to grow the tree including: list_of_features, feature_values, list_of_labels. DTFLModel ( model_name, model_spec, dt_model=None, latest_counts=None ) Ĭlass implementation for basic decision tree _init_ ( model_name, model_spec, dt_model=None, latest_counts=None, **kwargs ) ¶Ĭreate a 'DTFLModel' instance for decision tree model.
