FP64 fix (#1225)

Author: KulikovNikita

onedal/neighbors/neighbors.py

@@ -557,6 +557,7 @@ def _onedal_fit(self, X, y, queue):
         params = self._get_onedal_params(X, y)
         train_alg_regr = _backend.neighbors.regression.train
         train_alg_srch = _backend.neighbors.search.train
+
         if gpu_device:
             return train_alg_regr(policy, params, *to_table(X, y)).model
         return train_alg_srch(policy, params, to_table(X)).model
@@ -581,6 +582,7 @@ def _onedal_predict(self, model, X, params, queue):
             model = self._onedal_model
         else:
             model = self._create_model(backend)
+        params['fptype'] = 'float' if X.dtype == np.float32 else 'double'
         result = backend.infer(policy, params, model, to_table(X))
 
         return result
@@ -708,6 +710,8 @@ def _onedal_predict(self, model, X, params, queue):
             model = self._onedal_model
         else:
             model = self._create_model(_backend.neighbors.search)
+
+        params['fptype'] = 'float' if X.dtype == np.float32 else 'double'
         result = _backend.neighbors.search.infer(policy, params, model, to_table(X))
 
         return result