Democratising AI
Project description
Gradsflow
An AutoML Library made with Optuna and PyTorch Lightning
Image Classification
from flash.core.data.utils import download_data
from flash.image import ImageClassificationData
from gradsflow.autoclassifier import AutoImageClassifier
# 1. Create the DataModule
download_data("https://pl-flash-data.s3.amazonaws.com/hymenoptera_data.zip", "./data")
datamodule = ImageClassificationData.from_folders(
train_folder="data/hymenoptera_data/train/",
val_folder="data/hymenoptera_data/val/",
)
model = AutoImageClassifier(
datamodule, max_epochs=2, optimization_metric="val_accuracy"
)
print("AutoImageClassifier initialised!")
model.fit()
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
gradsflow-0.0.1a0.tar.gz
(16.5 kB
view hashes)
Built Distribution
Close
Hashes for gradsflow-0.0.1a0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 91dc7c68bcf1e9b6624b8fd3c3eeeb9cdd739c3cb905592373704d5ec8606bcd |
|
MD5 | 1c9c70f1ad0799fddf0e668fa2c684d6 |
|
BLAKE2b-256 | dc6893f58aad22bc86d55ce2c561821d3382aa4209b2d9fe1642409fb1ba8c44 |