Difference in prediction values for a binary classifier in ML.net and Tensorflow

127 views Asked by At

I was comparing results between the same model generated and used between two frameworks (Ml.net and tensorflow/keras) independently.

These are the steps in followed in getting the results:

  1. Train Model in Keras save as .h5 format.
  2. Convert to .pb model to use in ML.net, as ml.net currently uses tensorflow graphs and not saved model format.

Now for the same two images, the prediction value should be the same. As the model is the same. The difference being the underlying framework.

I could find only bits and pieces regarding why this occours. I am running both the programs with same underlying hardware and software configurations.

Is the underlying cause the difference in framework?

The image prediction scores for example images :

ml.net : 0.9929248 Fail keras/tensorflow : 0.99635077 Fail

The difference is minute, but being deterministic irrespective of platform is a feature that I am looking for.

0

There are 0 answers