neuraxle – Tarik Billa

Will scikit-learn utilize GPU?

December 30, 2022 by Tarik

Tensorflow only uses GPU if it is built against Cuda and CuDNN. By default it does not use GPU, especially if it is running inside Docker, unless you use nvidia-docker and an image with a built-in support. Scikit-learn is not intended to be used as a deep-learning framework and it does not provide any GPU … Read more

What is exactly sklearn.pipeline.Pipeline?

November 22, 2022 by Tarik

Transformer in scikit-learn – some class that have fit and transform method, or fit_transform method. Predictor – some class that has fit and predict methods, or fit_predict method. Pipeline is just an abstract notion, it’s not some existing ml algorithm. Often in ML tasks you need to perform sequence of different transformations (find set of … Read more