Python test_keras_model Beispiele

Beispiel #1

  def testTargetOps(self):
    model = self._getSingleInputKerasModel()
    keras_file = self._saveKerasModel(model)

    model_coverage.test_keras_model(
        keras_file,
        target_ops=set([lite.OpsSet.TFLITE_BUILTINS,
                        lite.OpsSet.SELECT_TF_OPS]))

Beispiel #2

Datei: model_coverage_lib_test.py Projekt: Wajih-O/tensorflow

  def testTargetOps(self):
    model = self._getSingleInputKerasModel()
    keras_file = self._saveKerasModel(model)

    model_coverage.test_keras_model(
        keras_file,
        target_ops=set([lite.OpsSet.TFLITE_BUILTINS,
                        lite.OpsSet.SELECT_TF_OPS]))

Beispiel #3

Datei: model_coverage_lib_test.py Projekt: qwerzou1/shibie

    def testPostTrainingQuantize(self):
        model = self._getSingleInputKerasModel()
        keras_file = self._saveKerasModel(model)

        model_coverage.test_keras_model(keras_file,
                                        post_training_quantize=True)

Beispiel #4

Datei: model_coverage_lib_test.py Projekt: qwerzou1/shibie

    def testFloat(self):
        model = self._getSingleInputKerasModel()
        keras_file = self._saveKerasModel(model)

        model_coverage.test_keras_model(keras_file)

Beispiel #5

Datei: model_coverage_lib_test.py Projekt: Wajih-O/tensorflow

  def testPostTrainingQuantize(self):
    model = self._getSingleInputKerasModel()
    keras_file = self._saveKerasModel(model)

    model_coverage.test_keras_model(keras_file, post_training_quantize=True)

Beispiel #6

Datei: model_coverage_lib_test.py Projekt: Wajih-O/tensorflow

  def testFloat(self):
    model = self._getSingleInputKerasModel()
    keras_file = self._saveKerasModel(model)

    model_coverage.test_keras_model(keras_file)