romanbredehoft-zama commited on
Commit
32a7fbc
1 Parent(s): 9aaa627

Remove irrelevant mention of XGB model

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. utils/model.py +2 -2
app.py CHANGED
@@ -310,8 +310,8 @@ with demo:
310
  Once the server receives the encrypted inputs, it can compute the prediction without ever
311
  needing to decrypt any value.
312
 
313
- This server employs an [XGBoost](https://github.com/dmlc/xgboost) classifier model that has
314
- been trained on a synthetic data-set.
315
  """
316
  )
317
 
 
310
  Once the server receives the encrypted inputs, it can compute the prediction without ever
311
  needing to decrypt any value.
312
 
313
+ This server employs a [Decision Tree](https://scikit-learn.org/stable/modules/generated/sklearn.tree.DecisionTreeClassifier.html)
314
+ classifier model that has been trained on a synthetic data-set.
315
  """
316
  )
317
 
utils/model.py CHANGED
@@ -110,8 +110,8 @@ class MultiInputModel:
110
  assert self._tree_inference is not None, self._is_not_fitted_error_message()
111
 
112
  if not self._is_compiled:
113
- xgb_inference = self._tree_inference
114
- self._tree_inference = lambda *args: xgb_inference(numpy.concatenate(args, axis=1))
115
 
116
  input_names = [f"input_{i}_encrypted" for i in range(len(inputs_encryption_status))]
117
 
 
110
  assert self._tree_inference is not None, self._is_not_fitted_error_message()
111
 
112
  if not self._is_compiled:
113
+ tree_inference = self._tree_inference
114
+ self._tree_inference = lambda *args: tree_inference(numpy.concatenate(args, axis=1))
115
 
116
  input_names = [f"input_{i}_encrypted" for i in range(len(inputs_encryption_status))]
117