Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Keras Implementation of Deepmind's WaveNet for Supervised Learning Tasks

Notifications You must be signed in to change notification settings

mjpyeon/wavenet-classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 

Repository files navigation

wavenet-classifier

Modified WaveNet Architecture for Supervised Learning Tasks

The goal of this project is to provide a model for speech classification and/or regression using WaveNet architecture which is originally designed as a generative model. So, this project provides an API for various supervised learning tasks related to speechs. Note that our implementation is based on keras2 on tensorflow background.

Usage


>>> from WaveNetClassifier import WaveNetClassifier

>>> wnc = WaveNetClassifier((96000,), (10,), kernel_size = 2, dilation_depth = 9, n_filters = 40, task = 'classification')
>>> wnc.fit(X_train, y_train, validation_data = (X_val, y_val), epochs = 100, batch_size = 32, optimizer='adam', save=True, save_dir='./')
>>> y_pred = wnc.predict(X_test)