Constraining the output interval of the GaussianLSTMModel to [0.0 .. 1.0]

[glisca]

What would be the simplest way to constrain the GaussianLSTMModel [1] to output values only within the interval [0.0 .. 1.0]?

[1]

garage/src/garage/tf/models/gaussian_lstm_model.py

Line 16 in 6461a07

class GaussianLSTMModel(Model):

Many thanks in advance!

[krzentner]

I don't believe this is possible with the current API, but it is not difficult to implement.
My recommendation is to copy gaussian_lstm_model.py into your project's source, and modify line 260 to wrap a TransformedDistribution with a Sigmoid, around dist.

A PR implementing this (gated by a flag) would of course also be welcome.