williamstoto/imo

A model for improving out-of-domain generalization by learning invariant features.

IMO

Table of Contents

• Setup
• Usage

Setup

We use conda to manage python packages.

git clone https://github.com/yourusername/IMO.git
cd IMO
conda env create -n IMO --file environment.yml
conda activate IMO

Usage

Training and Evaluate Models

Run the following command to start the training process:

# train a model on TweetEval dataset
sbatch submit_job_example.sh

# evaluate model on IMDB dataset
python run_text_disentangled_classification.py --train_file dataset/sentiment/original_imdb/test.csv --validation_file dataset/sentiment/original_imdb/test.csv --model_name_or_path checkpoints/your_model  --output_dir checkpoints/out/ --model_class BartForTokenAttentionSparseCLSJoint_incremental --only_evaluation

imo's Issues

Missing some class end with "_incremental"

Hello. Very interesting article. I want to test them on my data but I see that in your code there are a few classes that don't seem to be implemented yet.