Skip to content

Commit a043e86

Browse files
committed
README: añadir enlace proyecto W&B y resultados de mejor run individual
1 parent 0ec82d7 commit a043e86

1 file changed

Lines changed: 5 additions & 34 deletions

File tree

README.md

Lines changed: 5 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -2,35 +2,6 @@
22

33
Clasificador de pistachos (Kirmizi vs Siirt) usando Deep Learning con PyTorch Lightning y MLOps.
44

5-
## Estructura del proyecto
6-
7-
```
8-
.
9-
├── config/
10-
│ └── configuracion.yaml # Hiperparametros del modelo
11-
├── src/
12-
│ ├── main.py # Entrypoint CLI (train/sweep)
13-
│ ├── train.py # Entrenamiento con W&B logging
14-
│ ├── sweep.py # W&B Sweeps (grid search)
15-
│ ├── model.py # Modelos CNN (LightningModules)
16-
│ ├── data_module.py # LightningDataModule (kagglehub)
17-
│ ├── api_inferencia.py # FastAPI (POST /predict)
18-
│ ├── utils.py # load_config, get_project_root
19-
│ └── logging_config.py # Configuracion de logging
20-
├── tests/
21-
│ ├── test_api.py
22-
│ ├── test_data_module.py
23-
│ ├── test_model.py
24-
│ └── test_utils.py
25-
├── notebook/
26-
│ └── pistachio.ipynb # Notebook exploratorio
27-
├── Dockerfile # Containerizacion (MODE=train|api)
28-
├── entrypoint.sh # Entrypoint flexible
29-
├── .github/workflows/ci.yml # CI/CD con GitHub Actions
30-
├── requirements.txt
31-
└── pytest.ini
32-
```
33-
345
## Dataset
356

367
[Pistachio Image Dataset](https://www.kaggle.com/datasets/muratkokludataset/pistachio-image-dataset) - 2 clases (Kirmizi_Pistachio, Siirt_Pistachio). Se descarga automaticamente via kagglehub.
@@ -124,13 +95,13 @@ Respuesta:
12495
## W&B Report
12596

12697
- Proyecto: https://wandb.ai/14farresa-/pistachio-mlops
127-
- Sweep: https://wandb.ai/14farresa-/pistachio-mlops/sweeps/2147j105
98+
- Report: https://api.wandb.ai/links/14farresa-/9smbs51w
12899

129-
El proyecto incluye un W&B Report con:
100+
### Mejor run individual
130101

131-
- Resultados del sweep (learning rate, batch size, modelo)
132-
- Comparacion de arquitecturas (BatchNorm vs Dropout)
133-
- Metricas finales (accuracy, F1, matriz de confusion)
102+
| Modelo | LR | Batch | Test Acc | Test F1 |
103+
| ------------- | ------ | ----- | -------- | ------- |
104+
| cnn_batchnorm | 0.001 | 64 | 0.9317 | 0.9294 |
134105

135106
### Resultados del sweep
136107

0 commit comments

Comments
 (0)