Python cumlTSVD 예제들

예제 #1

파일: tsvd.py 프로젝트: tusharkalecam/cuml

    def _func_create_model(sessionId, dfs, **kwargs):
        try:
            from cuml.decomposition.tsvd_mg import TSVDMG as cumlTSVD
        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 cumlTSVD(handle=handle, **kwargs), dfs

예제 #2

 def _create_tsvd(handle, datatype, **kwargs):
     from cuml.decomposition.tsvd_mg import TSVDMG as cumlTSVD
     return cumlTSVD(handle=handle, output_type=datatype, **kwargs)