/StableReprLearning

Stable Representation Learning for Neural Network

Primary LanguagePython

StableReprLearning

This repository is the official implementation of Stable Representation Learning for Neural Network.

Requirements

To install requirements:

pip install -r requirements.txt

Run Experiments

To train and evaluate the model(s) with StableReprLearning framework in the paper, run this command:

python -m entry.srl_framework.<model> <number_of_experiments: 15> <early_stop_patience: 10>

To train and evaluate the model(s) with Normal framework, run this command:

python -m entry.normal_framework.<model> <number_of_experiments: 15> <early_stop_patience: 10>