Sleeplesss With scikit-learn

Machine Learning Psychic

Mi'kail Eli'yah
10 min readJun 11, 2024

Scikit-learn is a machine learning library built on NumPy, SciPy, and matplotlib. It’s focus is for data mining and data analysis. It includes various algorithms for classification, regression, clustering, and dimensionality reduction.

Usually, we include other machine learning libraries as well.

PyTorch is an open-source machine learning library developed by Facebook’s AI Research lab. It’s used for applications such as natural language processing and computer vision. 1 of the key features include Dynamic Computational Graphs, graphs (define-by-run), which makes it flexible and easy to debug. PyTorch’s core data structure (Tensors), similar to NumPy arrays, but with GPU acceleration. 1 of the advantages is that it has Autograd (Automatic differentiation for gradient-based optimization. auto_diff is available in tensorflow).

PyTorch Lightning is a lightweight wrapper on top of PyTorch that simplifies the process of building and training complex models. It is designed to make PyTorch more user-friendly and scalable. As speed performance is desired, it scales to facilitate easy scaling to multiple GPUs, TPUs, and across multiple nodes. There are boilerplate code for training, validation, and testing loops for Automated Training Loop.