ssoi / MDN

Experimenting with Mixture Density Network implemented in RStudio's TensorFlow API

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Mixture Density Network with TensorFlow (R API)

Intro


Implementation of Chris Bishop's Mixture Density Networks based on an inspiring blog post and RStudio's recently released TensorFlow API.

Results (Preliminary)

Using a relatively simple test case, estimating x = 7sin(3y/4) + y/2 (with Gaussian noise), TensorFlow successfully converges and fits the model. An example of sample (B=1000) drawn from the predicted distribution of y given a random test sample is shown: Prediction

TODO

  • Employ magrittr %>% paradigm to make construction of TensorFlow model easier to read
  • Move code for constructing TensorFlow model into functions
  • Create a Jupyter notebook with illustrative code
  • Find a more interesting example!
  • Explore extensions of model
  • Higher-dimension covariates
  • More complex neural architecture
  • Multivariate Gaussian (non-diagonal covariance)

About

Experimenting with Mixture Density Network implemented in RStudio's TensorFlow API


Languages

Language:R 100.0%