Ejemplo n.º 1
0
    def _func_create_model(sessionId, **kwargs):
        try:
            from cuml.neighbors.kneighbors_classifier_mg import \
                KNeighborsClassifierMG as cumlKNN
        except ImportError:
            raise_mg_import_exception()

        handle = worker_state(sessionId)["handle"]
        return cumlKNN(handle=handle, **kwargs)
Ejemplo n.º 2
0
    def _func_create_model(sessionId, **kwargs):
        try:
            from cuml.neighbors.kneighbors_regressor_mg import \
                KNeighborsRegressorMG as cumlKNN
        except ImportError:
            raise_mg_import_exception()

        handle = get_raft_comm_state(sessionId)["handle"]
        return cumlKNN(handle=handle, **kwargs)