Class MLInferenceEngineLSTMTests


  • public class MLInferenceEngineLSTMTests
    extends java.lang.Object
    • Constructor Detail

      • MLInferenceEngineLSTMTests

        public MLInferenceEngineLSTMTests()
    • Method Detail

      • beforeTest

        public static void beforeTest()
      • afterTest

        public static void afterTest()
      • testRunInferencesOnStatefulModel

        public void testRunInferencesOnStatefulModel()
        Compares the inference results on the LSTM MNIST Model from tflite examples. https://github.com/tensorflow/tflite-micro/tree/main/tensorflow/lite/micro/examples/mnist_lstm Without resetting the model, tests show that the digit 9 is predicted incorrectly as 3. When resetting the model, tests show that the digit 9 is predicted correctly as 9. So this test just runs the inferences twice and compares the results, there must be a different set of results.