infergo

infergo

infergo is a lightweight, easy-to-use library for running machine learning models in Go.

Installation

go get github.com/joeychilson/infergo

Example

Check out the examples directory for examples.

Available Models

• BERT - Text classification
• ResNet - Image classification
• YOLO - Object detection