Add readme & enviroments requirements.txt

This commit is contained in:
2026-06-30 15:39:33 +03:00
parent 8f095c7734
commit 4ab397c452
2 changed files with 107 additions and 0 deletions

16
requirements.txt Normal file
View File

@@ -0,0 +1,16 @@
# EdgeNeXt + Optuna HPO on Naruto Sign — pinned-ish minimums.
# Use a Python 3.103.12 env (PyTorch wheels are not yet built for 3.14).
torch>=2.2
torchvision>=0.17
timm>=1.0.7 # EdgeNeXt weights + Mixup/SoftTargetCrossEntropy
optuna>=4.5 # AutoSampler / multi-objective / pruners
optunahub>=0.2 # AutoSampler lives here (load_module("samplers/auto_sampler"))
umap-learn>=0.5.5
scikit-learn>=1.3
hdbscan>=0.8.33 # optional, density clustering on UMAP mid-dim
matplotlib>=3.7
numpy>=1.24
pillow>=10.0
pandas>=2.0
kagglehub>=0.3 # optional, dataset download
plotly>=5.18 # optional, optuna.visualization interactive plots