Update perplexity evaluator

* Introduce results channel
* Introduce jobs channel
* Generalize evaluation via callback

1 files changed, 0 insertions, 70 deletions

diff --git a/llm/cmd/ppl/ppl.go b/llm/cmd/ppl/ppl.go
deleted file mode 100644
index 6912e48..0000000
--- a/llm/cmd/ppl/ppl.go
+++ /dev/null
@@ -1,70 +0,0 @@
-package main
-
-import (
-	"fmt"
-	"log"
-
-	"github.com/jonasknobloch/mbpe"
-
-	"go.jknobloc.com/x/dataset"
-	"go.jknobloc.com/x/gpt2"
-	"go.jknobloc.com/x/llm"
-)
-
-func main() {
-	d := data()
-	m := model()
-	t := tokenizer()
-
-	e := llm.NewEvaluator()
-
-	e.SetTokenizer(t)
-	e.AddModel(m)
-
-	ppl, err := e.Perplexity(d, 1024, 512)
-
-	if err != nil {
-		log.Fatal(err)
-	}
-
-	fmt.Println(ppl)
-}
-
-func data() *dataset.ParquetReader {
-	var miniPile *dataset.ParquetReader
-
-	if r, err := dataset.NewParquetReader("dataset/cmd/dataset/tmp/minipile/validation"); err != nil {
-		log.Fatal(err)
-	} else {
-		miniPile = r
-	}
-
-	return miniPile
-}
-
-func model() *gpt2.Model {
-	m := gpt2.NewModel("gpt2/models/base/model.onnx", "0")
-
-	if err := m.Init(); err != nil {
-		log.Fatal(err)
-	}
-
-	return m
-}
-
-func tokenizer() *mbpe.Tokenizer {
-	m := mbpe.NewMBPE()
-
-	if err := m.Load("gpt2/models/base/vocab.json", "gpt2/models/base/merges.txt"); err != nil {
-		log.Fatal(err)
-	}
-
-	t := mbpe.NewTokenizer(m)
-
-	byteLevel := mbpe.NewByteLevel(false)
-
-	t.SetPreTokenizer(byteLevel)
-	t.SetDecoder(byteLevel)
-
-	return t
-}