Join TensorFlow at Google I/O, May 11-12 Register now

Module: tflite_model_maker.audio_classifier

APIs to train an audio classification model.


Demo code:


class AudioClassifier: Audio classifier for training/inference and exporing.

class BrowserFftSpec: Model good at detecting speech commands, using Browser FFT spectrum.

class DataLoader: DataLoader for audio tasks.

class YamNetSpec: Model good at detecting environmental sounds, using YAMNet embedding.


create(...): Loads data and retrains the model.