danf0 commited on
Commit
d463fa5
1 Parent(s): 167906f

Add citation information.

Browse files
Files changed (1) hide show
  1. README.md +10 -1
README.md CHANGED
@@ -28,6 +28,8 @@ VS(K) = exp(tr(K/n @ log(K/n))) = exp(-sum_i lambda_i log lambda_i),
28
  where `lambda_i` are the eigenvalues of `K/n` and `0 log 0 = 0`.
29
  That is, the Vendi Score is equal to the exponential of the von Neumann entropy of `K/n`, or the Shannon entropy of the eigenvalues, which is also known as the effective rank.
30
 
 
 
31
  ## How to Use
32
  The Vendi Score is available as a Python package or in HuggingFace `evaluate`.
33
  To use the Python package, see the instructions at https://github.com/vertaix/Vendi-Score.
@@ -124,4 +126,11 @@ N-grams: 3.91, BERT: 1.21
124
  The Vendi Score depends on the choice of similarity function. Care should be taken to select a similarity function that reflects the features that are relevant for defining diversity in a given application.
125
 
126
  ## Citation
127
-
 
 
 
 
 
 
 
 
28
  where `lambda_i` are the eigenvalues of `K/n` and `0 log 0 = 0`.
29
  That is, the Vendi Score is equal to the exponential of the von Neumann entropy of `K/n`, or the Shannon entropy of the eigenvalues, which is also known as the effective rank.
30
 
31
+ For more information, please see our paper, [The Vendi Score: A Diversity Evaluation Metric for Machine Learning](https://arxiv.org/abs/2210.02410).
32
+
33
  ## How to Use
34
  The Vendi Score is available as a Python package or in HuggingFace `evaluate`.
35
  To use the Python package, see the instructions at https://github.com/vertaix/Vendi-Score.
 
126
  The Vendi Score depends on the choice of similarity function. Care should be taken to select a similarity function that reflects the features that are relevant for defining diversity in a given application.
127
 
128
  ## Citation
129
+ ```bibtex
130
+ @article{friedman2022vendi,
131
+ title={The Vendi Score: A Diversity Evaluation Metric for Machine Learning},
132
+ author={Friedman, Dan and Dieng, Adji Bousso},
133
+ journal={arXiv preprint arXiv:2210.02410},
134
+ year={2022}
135
+ }
136
+ ```