diff options
Diffstat (limited to 'llm/perplexity.go')
| -rw-r--r-- | llm/perplexity.go | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/llm/perplexity.go b/llm/perplexity.go index 9596e8d..2240f2d 100644 --- a/llm/perplexity.go +++ b/llm/perplexity.go @@ -14,7 +14,7 @@ import ( "go.jknobloc.com/x/dataset" ) -func (e *Evaluator) Perplexity(data *dataset.Reader, window, stride int) (float64, error) { +func (e *Evaluator) Perplexity(data *dataset.ParquetReader, window, stride int) (float64, error) { docs := make([]string, 0) n := 0 |