romanbredehoft-zama
commited on
Commit
•
32a7fbc
1
Parent(s):
9aaa627
Remove irrelevant mention of XGB model
Browse files- app.py +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
|
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 |
-
|
114 |
-
self._tree_inference = lambda *args:
|
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 |
|