This repository provides an implementation of an ensemble learning framework for classification tasks. Below are the instructions to set up the environment, download the saved model weights, and run the framework.
The saved model weights are stored on Google Drive. Follow these steps to download and set them up:
- Download the weights from Google Drive Link.
- Create a directory named
saved_weights
in the root folder of this project. - Place the downloaded weights inside the
saved_weights
folder.
To avoid dependency conflicts, we recommend setting up a Python virtual environment. Follow these steps:
conda create -n myenv python=3.7
conda activate myenv
pip install pytorch-ignite==0.4.2
pip install dgl==0.6.0 -f https://data.dgl.ai/wheels-test/repo.html
pip install numpy==1.17.2
pip install torch==1.11.0
pip install transformers==4.19.2
pip install nltk==3.4.5
pip install scikit-learn==0.22
pip install numpy==1.17.2 torch==1.11.0 transformers==4.19.2
python ensemble_learning.py