We're excited to announce that we're launching the second installment of Dataiku Product Days Register Now

Distributed Training Machine Learning

Level 2
Distributed Training Machine Learning


What I enjoy about Dataiku is the visual machine learning.

I have a 21 GB Dataset to train and I'd like to try it on Dataiku with XGBOOST however it will take a while.

I have a couple machines that connect in a SSH cluster.

Is there anyway I can create a Dask SSH cluster in Dataiku so I can use the visual machine learning to train the data?

In my jupyter notebook, I create the SSH dask cluster as follows:

from dask.distributed import Client, SSHCluster
cluster = SSHCluster(
        ["localhost", "", ""],
        connect_options={"known_hosts": None,"username": "vinhdiesal"},
        worker_options={"nthreads": 20, "local_directory":"/tmp/"},
        scheduler_options={"port": 0, "dashboard_address": ":8797"},
        worker_module= 'dask_cuda.dask_cuda_worker'
client = Client(cluster)



0 Kudos
1 Reply


Thanks for the positive feedback about visual ML in DSS. However, I have to admit, that Dask isn't integrated into it in any way. The only way you could proceed in DSS is by using Notebooks and implementing the Dask interaction yourself.


Andrey Avtomonov
R&D Engineer @ Dataiku
0 Kudos
A banner prompting to get Dataiku DSS