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)
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)