Predictor - improve dimension check from ONNX

tf-encrypted / moose

Secure distributed dataflow framework for encrypted machine learning and data processing

Apache License 2.0

58 stars 16 forks source link

Closed yanndupis closed 2 years ago

yanndupis commented 2 years ago

Check if the number of coefficients matches the input dimension in the ONNX file for linear and logistic regression models.