Spaces:
Sleeping
Sleeping
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 |
|