search

tolstikhin / wae

Wasserstein Auto-Encoders
BSD 3-Clause "New" or "Revised" License
505 stars 90 forks source link

One Dimensional WAE #12

Open payman-ap opened 5 years ago

payman-ap commented 5 years ago

Hi,

Thanks for your shared codes and I learned so much out of it!

I'm going to modify your wae implementation for 1D sequential data (protein sequences reconstruction), However, your codes are hugely image specific. Can you help me with that?!

tolstikhin commented 5 years ago

Hey!

Thanks for your interest. Yes indeed, at this point the input pipeline is pretty much image-specific.

One way to go about it is to reshape 1d array into (1, N, 1) and feed that to the fully connected / convolutional network architectures with proper kernel size.