diff options
Diffstat (limited to 'tensorflow_lite_support/cc/task/text/nlclassifier/nl_classifier.h')
-rw-r--r-- | tensorflow_lite_support/cc/task/text/nlclassifier/nl_classifier.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow_lite_support/cc/task/text/nlclassifier/nl_classifier.h b/tensorflow_lite_support/cc/task/text/nlclassifier/nl_classifier.h index 2a9573a1..05189dda 100644 --- a/tensorflow_lite_support/cc/task/text/nlclassifier/nl_classifier.h +++ b/tensorflow_lite_support/cc/task/text/nlclassifier/nl_classifier.h @@ -112,6 +112,9 @@ class NLClassifier : public core::BaseTaskApi<std::vector<core::Category>, // Performs classification on a string input, returns classified results. std::vector<core::Category> Classify(const std::string& text); + // Gets the model version, or "NO_VERSION_INFO" in case there is no version. + std::string GetVersion() const; + protected: static constexpr int kOutputTensorIndex = 0; static constexpr int kOutputTensorLabelFileIndex = 0; |