Skip to content

Commit 7a22a2c

Browse files
authored
Update README.md
1 parent 8528b50 commit 7a22a2c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
![](assets/banner.png)
88

9-
This repository contains the code to reproduce [Deep Evidential Regression](http://www.mit.edu/~amini/pubs/pdf/deep-evidential-regression.pdf), as published in [NeurIPS 2020](https://neurips.cc/), as well as more general code to leverage evidential learning to train neural networks to learn their own measures of uncertainty directly from data!
9+
This repository contains the code to reproduce [Deep Evidential Regression](https://proceedings.neurips.cc/paper/2020/file/aab085461de182608ee9f607f3f7d18f-Paper.pdf), as published in [NeurIPS 2020](https://neurips.cc/), as well as more general code to leverage evidential learning to train neural networks to learn their own measures of uncertainty directly from data!
1010

1111
## Setup
1212
To use this package, you must install the following dependencies first:
@@ -24,7 +24,7 @@ Now you're ready to start using this package directly as part of your existing `
2424
```
2525

2626
### Example
27-
To use evidential deep learning, you must edit the last layer of your model to be *evidential* and use a supported loss function to train the system end-to-end. This repository supports evidential layers for both fully connected and convolutional (2D) layers. The evidential prior distribution presented in the paper follow a Normal Inverse-Gamma and can be added to your model:
27+
To use evidential deep learning, you must edit the last layer of your model to be *evidential* and use a supported loss function to train the system end-to-end. This repository supports evidential layers for both fully connected and convolutional (2D) layers. The evidential prior distribution presented in the paper follows a Normal Inverse-Gamma and can be added to your model:
2828

2929
```
3030
import evidential_deep_learning as edl

0 commit comments

Comments
 (0)