DATASET EXPLORER & SKELETON DIRECTORY
Explore ready-to-train datasets with automated PyTorch DataLoaders, feature statistics, class distributions, and 7-step guided workspace pipelines.
Titanic: Machine Learning from Disaster
Predict passenger survival on the Titanic using passenger class, sex, age, ticket fare, cabin, and family size with PyTorch Multi-Layer Perceptron and ROC-AUC evaluation.
MNIST Handwritten Digits
The standard computer vision benchmark dataset of handwritten digits (0-9) to master PyTorch Convolutional Neural Networks (Conv2d, MaxPool2d, BatchNorm2d).
CIFAR-10 Object Recognition
Natural RGB image recognition benchmark dataset for training deep residual networks (ResNet-18) and real-time image augmentation pipelines in PyTorch.
California Housing Prices
Predict median district housing values across California based on median income, housing age, average rooms, population, and latitude/longitude coordinates.
IMDb Large Movie Review Dataset
Binary natural language sentiment classification benchmark with 25,000 training and 25,000 testing movie reviews using PyTorch word embeddings and Bidirectional LSTMs.
Credit Card Fraud Detection
Highly imbalanced anonymized financial transactions requiring Focal Loss, weighted cross-entropy, Precision-Recall AUC tracking, and anomaly detection in PyTorch.
Fashion-MNIST Clothing Classification
A drop-in replacement for MNIST containing 10 categories of Zalando clothing articles with higher intra-class visual complexity and spatial textures.
Wine Quality (Red & White)
Physicochemical laboratory tests (fixed acidity, volatile acidity, citric acid, residual sugar, chlorides, alcohol) to predict wine sensory quality scores.
Heart Disease UCI (Cleveland)
Clinical cardiology database to predict coronary artery disease presence from resting blood pressure, cholesterol, fasting blood sugar, and exercise ST-depression.
Breast Cancer Wisconsin Diagnostic
Computed nuclear morphometry features from fine needle aspirate (FNA) biopsy images (radius, texture, perimeter, area, smoothness, concavity, symmetry).
Iris Flower Dataset
The classic multi-class pattern recognition dataset for learning PyTorch softmax distributions, one-hot target encoding, and multi-class decision boundaries.
Reuters-21578 News Wire Topics
Multi-label text classification of financial and economic news articles with class imbalances using PyTorch tokenization and BCEWithLogitsLoss.
Bike Sharing Demand
Forecast hourly bike rental demand using weather, seasonality, holiday indicators, and cyclical temporal encodings (sin/cos hour transformations).
Palmer Archipelago Penguins
Modern tabular dataset featuring culmen dimensions, flipper length, body mass, and island origins with missing value imputation and categorical encoding in PyTorch.
SMS Spam Collection
Detect fraudulent SMS spam messages using subword tokenization, vocabulary mapping, PyTorch EmbeddingBag, and imbalanced cross-entropy loss.
Diamonds Dataset
Predict the price of round cut diamonds based on carat weight, cut quality, color grade, clarity rating, and physical spatial dimensions (x, y, z).
Bank Marketing (Direct Telemarketing)
Predict whether a retail banking client will subscribe a long-term deposit based on campaign contact history and macroeconomic financial indicators.
CelebA Facial Attributes Mini
Predict multiple facial binary attributes simultaneously using deep multi-task convolutional neural network architectures in PyTorch.
Steel Plates Faults (UCI)
Classify surface defects and structural faults in stainless steel plates using 27 geometric, thickness, and luminosity sensor measurements.
COCO Mini Object Detection & BBoxes
Hands-on PyTorch bounding box coordinate regression, Complete IoU (CIoU) loss calculation, and non-maximum suppression (NMS) evaluation.
AG News 4-Class Topic Classification
Academic benchmark for multi-class news article topic modeling with word tokenization, PyTorch BiLSTM, 1D-CNN, and Macro-F1 evaluation.
Rotten Tomatoes Critic Reviews
Sentence-level movie review sentiment analysis with nuanced linguistic phrasing, idioms, and negation handling via PyTorch Recurrent Networks.
Emotion Tweet & Text Classifier
Multi-class affective state detection classifying subtle human psychological emotions from informal digital text using PyTorch LSTMs.
Oxford-IIIT Pet Breed Classification
Fine-grained visual classification of 37 domestic dog and cat breeds with subtle inter-class visual similarities using transfer learning in PyTorch.
Medical Blood Cell Microscope Vision
Automated clinical hematology cytology classification for white blood cell differential counts using deep convolutional neural networks in PyTorch.
Food-101 Mini Cuisine Image Classifier
Visual culinary dish recognition with color jitter, random crops, and transfer learning with MobileNetV2 in PyTorch.
Tiny Shakespeare Playwright Generator
Autoregressive sequential language modeling to generate dramatic Elizabethan dialogue using PyTorch Character/Word LSTMs and temperature sampling.
Edgar Allan Poe Gothic Poetry Generator
Autoregressive language model synthesizing dark romantic poetry and gothic prose mimicking Poe's vocabulary and rhythmic cadence in PyTorch.
Python Code & Docstring Auto-Complete
Causal code language model predicting Python AST expressions, docstrings, and function bodies with PyTorch Recurrent / Causal Transformer models.