NonEiV training included
This covers all regression datasets treated in the MC Dropout and Deep Ensemble paper. Results are comparable or even better. For multivariate dataset, the decouple_dimensions keyword in the evaluation scripts can be used to follow the (rather weird) convention of these papers.
Showing
- Experiments/evaluate_california.py 58 additions, 0 deletionsExperiments/evaluate_california.py
- Experiments/evaluate_energy.py 1 addition, 1 deletionExperiments/evaluate_energy.py
- Experiments/evaluate_kin8nm.py 1 addition, 1 deletionExperiments/evaluate_kin8nm.py
- Experiments/evaluate_msd.py 1 addition, 1 deletionExperiments/evaluate_msd.py
- Experiments/evaluate_naval.py 1 addition, 1 deletionExperiments/evaluate_naval.py
- Experiments/evaluate_power.py 1 addition, 1 deletionExperiments/evaluate_power.py
- Experiments/evaluate_protein.py 1 addition, 1 deletionExperiments/evaluate_protein.py
- Experiments/evaluate_wine.py 1 addition, 1 deletionExperiments/evaluate_wine.py
- Experiments/evaluate_yacht.py 1 addition, 1 deletionExperiments/evaluate_yacht.py
- Experiments/train_noneiv_california.py 2 additions, 2 deletionsExperiments/train_noneiv_california.py
- Experiments/train_noneiv_concrete.py 2 additions, 2 deletionsExperiments/train_noneiv_concrete.py
- Experiments/train_noneiv_energy.py 2 additions, 2 deletionsExperiments/train_noneiv_energy.py
- Experiments/train_noneiv_kin8nm.py 2 additions, 2 deletionsExperiments/train_noneiv_kin8nm.py
- Experiments/train_noneiv_msd.py 2 additions, 2 deletionsExperiments/train_noneiv_msd.py
- Experiments/train_noneiv_naval.py 2 additions, 2 deletionsExperiments/train_noneiv_naval.py
- Experiments/train_noneiv_power.py 2 additions, 2 deletionsExperiments/train_noneiv_power.py
- Experiments/train_noneiv_protein.py 2 additions, 2 deletionsExperiments/train_noneiv_protein.py
- Experiments/train_noneiv_wine.py 2 additions, 2 deletionsExperiments/train_noneiv_wine.py
- Experiments/train_noneiv_yacht.py 2 additions, 2 deletionsExperiments/train_noneiv_yacht.py
- README.md 2 additions, 2 deletionsREADME.md
Loading
Please register or sign in to comment