oramasearch / onnx-go

onnx-go gives the ability to import a pre-trained neural network within Go without being linked to a framework or library.
https://blog.owulveryck.info/2019/04/03/from-a-project-to-a-product-the-state-of-onnx-go.html
MIT License
704 stars 72 forks source link

Implement operator `LinearClassifier` for backend `Gorgonia` #174

Open senkrish opened 4 years ago

senkrish commented 4 years ago

Why is this operator needed?

I get this error when trying to run a backend from a model created using sciket_learn Python module (sklearn.linear_model.LogisticRegression).

onnx: operator LinearClassifier not implemented ()

Implementation

Link to existing material on the backend

N/A

Expected problems?

N/A

Tests

go test -run=ONNX/TestOperator

trams commented 4 years ago

I am also interested in this

owulveryck commented 4 years ago

Hi!

I will try to dig in this hopefully in two weeks. Many thanks for your interest in onnx-go