Skip to content

Comparison of Merchine Learning Libraries

JUNGSHIC PARK edited this page Dec 3, 2022 · 1 revision

Scikit-Learn (sklearn)

  • Open-source machine learning library for Python
  • Built on top of and integrates with commonly used libraries such as NumPy, SciPy, Matplotlib and pandas.
  • For Beginner and non-neural network algorithms, relatively small data sets.

TensorFlow

  • Specialized in deep learning and neural networks
  • Supports languages such as Python, C/C++, Java and Javascript, and others.
  • Hardware acceleration through GPUs and TPU(Tensor Processing Unit)s

PyTouch

  • Deep learning software library for Python, C++ and Julia
  • Neural networks with the ability to create custom models

Keras

  • High-level deep learning framework that abstracts away many of the low-level details
  • Reduced code complexity despite being a deep-learning framework
  • A high-level interface for TensorFlow, resulting in comparatively slower computations with less complexity.