Skip to content

Commit

Permalink
58 data mining (#65)
Browse files Browse the repository at this point in the history
* updated model training and added kaggle set

kaggle set has more than 5000 images but not every face from each image could be detected

* feat: trained and evaluated different models.

* feat: added models.

* feat: added number of samples.

* feat: added output.

* feat: added titles to plots.

* feat: added title to plots.

* feat: added the analysis of the pyfeat classifier.

* refactor: adapt to kaggle

* ran pyfeat detector

---------

Co-authored-by: LasseFrm <64502978+LasseFrm@users.noreply.github.com>
Co-authored-by: Jurek Sander <44340639+sanderjk5@users.noreply.github.com>
  • Loading branch information
3 people authored Dec 18, 2023
1 parent 81cebe3 commit cb0b516
Show file tree
Hide file tree
Showing 24 changed files with 43,510 additions and 62 deletions.
4 changes: 4 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -158,3 +158,7 @@ cython_debug/
# and can be added to the global gitignore or merged into this file. For a more nuclear
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
#.idea/


# data
model-training/data/
1 change: 1 addition & 0 deletions .pre-commit-config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ repos:
- id: check-xml
- id: check-yaml
- id: check-added-large-files
args: ['--maxkb=10000']
- id: trailing-whitespace

- repo: https://github.com/astral-sh/ruff-pre-commit
Expand Down
2 changes: 1 addition & 1 deletion .vscode/settings.json
Original file line number Diff line number Diff line change
Expand Up @@ -11,5 +11,5 @@
"python.testing.unittestEnabled": false,
"python.testing.pytestEnabled": true,
"jupyter.debugJustMyCode": false,
"python.analysis.typeCheckingMode": "basic"
"python.analysis.typeCheckingMode": "off"
}
1,226 changes: 1,226 additions & 0 deletions model-training/analyse_pyfeat.ipynb

Large diffs are not rendered by default.

5,212 changes: 5,212 additions & 0 deletions model-training/aus_combined.csv

Large diffs are not rendered by default.

986 changes: 986 additions & 0 deletions model-training/aus_diffusion_fer.csv

Large diffs are not rendered by default.

4,227 changes: 4,227 additions & 0 deletions model-training/aus_kaggle_set.csv

Large diffs are not rendered by default.

Binary file added model-training/best_model.pkl
Binary file not shown.
Binary file added model-training/best_model_wo_pca.pkl
Binary file not shown.
1,165 changes: 1,165 additions & 0 deletions model-training/data_analysis_and_preprocessing.ipynb

Large diffs are not rendered by default.

Binary file added model-training/fer_label_encoder.pkl
Binary file not shown.
Binary file added model-training/fer_model.pkl
Binary file not shown.
Binary file added model-training/fer_scaler.pkl
Binary file not shown.
8,690 changes: 8,629 additions & 61 deletions model-training/model_training.ipynb

Large diffs are not rendered by default.

713 changes: 713 additions & 0 deletions model-training/model_training_and_evaluation.ipynb

Large diffs are not rendered by default.

634 changes: 634 additions & 0 deletions model-training/nn_training_and_evaluation.ipynb

Large diffs are not rendered by default.

Binary file added model-training/pca.pkl
Binary file not shown.
Loading

0 comments on commit cb0b516

Please sign in to comment.