utc

Go Inference API for UTC (Universal Text Classification).

Installation

1. Install utc

   $ go get -u github.com/go-aie/utc

2. Install Paddle Inference Go API

Documentation

Check out the documentation.

Testing

Generate the inference model:

$ python3 cli/cli.py download

Run tests:

$ go test -v -race | grep -E 'go|Test'
=== RUN   TestBuildInputsWithPrompt
--- PASS: TestBuildInputsWithPrompt (0.00s)
=== RUN   TestPromptTokenizer_Encode
--- PASS: TestPromptTokenizer_Encode (0.01s)
=== RUN   TestUTC_Run
--- PASS: TestUTC_Run (2.89s)
ok      github.com/go-aie/utc   3.748s

License

MIT