Attend the Women in ML Symposium on December 7 Register now

Module: tflite_model_maker.audio_classifier

Stay organized with collections Save and categorize content based on your preferences.

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.