Home

Awesome

TensorFlow Lite text classification sample

Overview

This is an end-to-end example of movie review sentiment classification built with TensorFlow 2.0 (Keras API), and trained on IMDB dataset. The demo app processes input movie review texts, and classifies its sentiment into negative (0) or positive (1).

Model

See Text Classification with Movie Reviews for a step-by-step instruction of building a simple text classification model.

iOS app

Follow the steps below to build and run the sample iOS app.

Requirements

Build and run

  1. Clone the repository to your computer to get the demo application.

    git clone https://github.com/khurram18/TextClassafication.git
    
  2. Navigate to the cloned directory

    cd TextClassafication
    
  3. Open the TextClassification.xcworkspace in Xcode either by double clicking on it or using below command

    open TextClassification.xcworkspace
    

    This launches Xcode and opens the TextClassification project.

Additional Note

_Please do not delete the empty reference to the .tflite file after you clone the repo and open the project. The model reference will be fixed as the model file is downloaded when the application is built and run for the first time.

The pull request to add this example to official tensorflow examples repository is here