Implementation of decision tree and decision forest algorithms. Each of these algorithms will be used in order to identify six basic emotions from the human facial expressions (anger, disgust, fear, happiness, sadness and surprise) based on a labelled set of facial Action Units (AUs). AUs correspond to contractions of human facial muscles, which underlie each and every facial expression, including facial expressions of the six basic emotions.
Install dependency libraries by running
sudo apt-get update
sudo apt-get -y install python3-pip
sudo pip3 install pipreqs
sudo pip3 install -r requirements.txt
It will install pip3 alognside the following libraries: networkx, matplotlib, pandas, scipy, numpy.
You can test our implementation of decision trees and decision forest by inputing the path to the .mat file you want to test in the following way:
To test decision trees implementation: ''' python3 test.py PATH_TO_FILE tree '''
To test decision forest implementation: ''' python3 test.py PATH_TO_FILE forest '''
We provide a command-line based program. The default command, which will run the decision forest algorithm on the Data/cleandata_students.mat is:
python3 main.py
The program actually takes between 0 and 2 arguments
0 -> default version from above.
python3 main.py
1 -> either 'tree' or 'forest', which decides which algorithm you want to apply on the data. The algorithm will run on a single process.
python3 main.py tree
python3 main.py forest
2 -> the first argument is either tree or forest, as above. The second argument should be either 'single' or 'multi', which will decide whether the chosen algorithm will be run on a single process or on multiple processes for improved computational time. Note that the multiprocess version only seems to be working on Linux environments.
python3 main.py tree single
python3 main.py forest single
python3 main.py tree multi
pythone main.py forest multi
The program provides utilities for visualizing the decision tree built for Data/cleandata_students.mat. Just run
python3 main.py visualisation
Visualisation will create (if not already created) a directory called "plots" which will contain 2 images for each decision tree per emotion (one image shows the structure of the tree the other one is made such nodes don't overlap)
- Adrian Catana
- Andrei Isaila
- Cristian Matache
- Oana Ciocioman