PyTorch聽vs聽scikit-learn

Need advice about which tool to choose?Ask the StackShare community!

PyTorch

871
963
+ 1
33
scikit-learn

833
882
+ 1
32
Add tool

PyTorch vs scikit-learn: What are the differences?

What is PyTorch? A deep learning framework that puts Python first. PyTorch is not a Python binding into a monolothic C++ framework. It is built to be deeply integrated into Python. You can use it naturally like you would use numpy / scipy / scikit-learn etc.

What is scikit-learn? Easy-to-use and general-purpose machine learning in Python. scikit-learn is a Python module for machine learning built on top of SciPy and distributed under the 3-Clause BSD license.

PyTorch and scikit-learn can be primarily classified as "Machine Learning" tools.

"Developer Friendly" is the top reason why over 2 developers like PyTorch, while over 14 developers mention "Scientific computing" as the leading cause for choosing scikit-learn.

PyTorch and scikit-learn are both open source tools. scikit-learn with 36K GitHub stars and 17.6K forks on GitHub appears to be more popular than PyTorch with 29.6K GitHub stars and 7.18K GitHub forks.

Repro, Home61, and MonkeyLearn are some of the popular companies that use scikit-learn, whereas PyTorch is used by Suggestic, cotobox, and Depop. scikit-learn has a broader approval, being mentioned in 71 company stacks & 40 developers stacks; compared to PyTorch, which is listed in 21 company stacks and 46 developer stacks.

Decisions about PyTorch and scikit-learn

Pytorch is a famous tool in the realm of machine learning and it has already set up its own ecosystem. Tutorial documentation is really detailed on the official website. It can help us to create our deep learning model and allowed us to use GPU as the hardware support.

I have plenty of projects based on Pytorch and I am familiar with building deep learning models with this tool. I have used TensorFlow too but it is not dynamic. Tensorflow works on a static graph concept that means the user first has to define the computation graph of the model and then run the ML model, whereas PyTorch believes in a dynamic graph that allows defining/manipulating the graph on the go. PyTorch offers an advantage with its dynamic nature of creating graphs.

See more
Fabian Ulmer
Software Developer at Hestia | 3 upvotes 路 13.3K views

For my company, we may need to classify image data. Keras provides a high-level Machine Learning framework to achieve this. Specifically, CNN models can be compactly created with little code. Furthermore, already well-proven classifiers are available in Keras, which could be used as Transfer Learning for our use case.

We chose Keras over PyTorch, another Machine Learning framework, as our preliminary research showed that Keras is more compatible with .js. You can also convert a PyTorch model into TensorFlow.js, but it seems that Keras needs to be a middle step in between, which makes Keras a better choice.

See more
Xi Huang
Developer at University of Toronto | 8 upvotes 路 44.3K views

For data analysis, we choose a Python-based framework because of Python's simplicity as well as its large community and available supporting tools. We choose PyTorch over TensorFlow for our machine learning library because it has a flatter learning curve and it is easy to debug, in addition to the fact that our team has some existing experience with PyTorch. Numpy is used for data processing because of its user-friendliness, efficiency, and integration with other tools we have chosen. Finally, we decide to include Anaconda in our dev process because of its simple setup process to provide sufficient data science environment for our purposes. The trained model then gets deployed to the back end as a pickle.

See more
Get Advice from developers at your company using Private StackShare. Sign up for Private StackShare.
Learn More
Pros of PyTorch
Pros of scikit-learn
  • 11
    Easy to use
  • 9
    Developer Friendly
  • 8
    Easy to debug
  • 5
    Sometimes faster than TensorFlow
  • 18
    Scientific computing
  • 14
    Easy

Sign up to add or upvote prosMake informed product decisions

Cons of PyTorch
Cons of scikit-learn
  • 2
    Lots of code
  • 1
    Limited

Sign up to add or upvote consMake informed product decisions

What is PyTorch?

PyTorch is not a Python binding into a monolothic C++ framework. It is built to be deeply integrated into Python. You can use it naturally like you would use numpy / scipy / scikit-learn etc.

What is scikit-learn?

scikit-learn is a Python module for machine learning built on top of SciPy and distributed under the 3-Clause BSD license.

Need advice about which tool to choose?Ask the StackShare community!

What companies use PyTorch?
What companies use scikit-learn?
See which teams inside your own company are using PyTorch or scikit-learn.
Sign up for Private StackShareLearn More

Sign up to get full access to all the companiesMake informed product decisions

What tools integrate with PyTorch?
What tools integrate with scikit-learn?

Sign up to get full access to all the tool integrationsMake informed product decisions

Blog Posts

+14
11
2027
Dec 4 2019 at 8:01PM

Pinterest

+4
5
2901
+42
46
38948
What are some alternatives to PyTorch and scikit-learn?
TensorFlow
TensorFlow is an open source software library for numerical computation using data flow graphs. Nodes in the graph represent mathematical operations, while the graph edges represent the multidimensional data arrays (tensors) communicated between them. The flexible architecture allows you to deploy computation to one or more CPUs or GPUs in a desktop, server, or mobile device with a single API.
Keras
Deep Learning library for Python. Convnets, recurrent neural networks, and more. Runs on TensorFlow or Theano. https://keras.io/
Caffe2
Caffe2 is deployed at Facebook to help developers and researchers train large machine learning models and deliver AI-powered experiences in our mobile apps. Now, developers will have access to many of the same tools, allowing them to run large-scale distributed training scenarios and build machine learning applications for mobile.
MXNet
A deep learning framework designed for both efficiency and flexibility. It allows you to mix symbolic and imperative programming to maximize efficiency and productivity. At its core, it contains a dynamic dependency scheduler that automatically parallelizes both symbolic and imperative operations on the fly.
Torch
It is easy to use and efficient, thanks to an easy and fast scripting language, LuaJIT, and an underlying C/CUDA implementation.
See all alternatives
How developers use PyTorch and scikit-learn
Yonas Tesh uses
PyTorch

I used PyTorch when i was working on an AI application, image classification using deep learning.