def _func_create_model(sessionId, dfs, **kwargs):
try:
from cuml.decomposition.pca_mg import PCAMG as cumlPCA
except ImportError:
raise Exception("cuML has not been built with multiGPU support "
"enabled. Build with the --multigpu flag to"
" enable multiGPU support.")
handle = worker_state(sessionId)["handle"]
return cumlPCA(handle=handle, **kwargs), dfs
def _create_pca(handle, datatype, **kwargs):
from cuml.decomposition.pca_mg import PCAMG as cumlPCA
return cumlPCA(handle=handle, output_type=datatype, **kwargs)