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

[Gheorghe Lisca]

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!

[K.R. Zentner]

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.