Spaces:
Sleeping
Sleeping
File size: 612 Bytes
4a5bfab f6b961e e69e3a3 ea3fe1b e69e3a3 ea3fe1b e69e3a3 ea3fe1b e69e3a3 | 1 2 3 4 5 6 7 8 9 10 | # data_preparation/
Load and split the .npz data. Used by all training code and notebooks.
**prepare_dataset.py:** `load_all_pooled()`, `load_per_person()` for LOPO, `get_numpy_splits()` (XGBoost), `get_dataloaders()` (MLP). Cleans yaw/pitch/roll and EAR to fixed ranges. Face_orientation uses 10 features: head_deviation, s_face, s_eye, h_gaze, pitch, ear_left, ear_avg, ear_right, gaze_offset, perclos.
**data_exploration.ipynb:** EDA — stats, class balance, histograms, correlations.
You don’t run prepare_dataset directly; import it from `models.mlp.train`, `models.xgboost.train`, or the notebooks.
|