-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathinterface.py
More file actions
35 lines (29 loc) · 1.26 KB
/
Copy pathinterface.py
File metadata and controls
35 lines (29 loc) · 1.26 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
# interface.py - Interface module for WBC_Classifier_CNN training and inference
"""
This interface connects all core components of the WBC classifier project.
The grader will use these imports to train and evaluate your model.
Ensure all components work independently and follow correct naming.
"""
# ----------------------------
# Dataset and DataLoader
# ----------------------------
from dataset import WBCDataset as TheDataset
from dataset import get_dataloaders as the_dataloader
# ----------------------------
# CNN Model
# ----------------------------
from model import WBC_Classifier_CNN as TheModel
# ----------------------------
# Training Loop (Cross-validation)
# ----------------------------
from train import wbc_trainer as the_trainer
# ----------------------------
# Inference on Single Image or Batch, along with optional full statistical evaluation utility
# ----------------------------
from predict import wbc_predictor as the_predictor
from predict import predict_model # - Optional full evaluation utility; Evaluates the best saved model (from the respective fold) on the complete test dataset
# ----------------------------
# Config Parameters
# ----------------------------
from config import batch_size as the_batch_size
from config import epochs as total_epochs