Пример слоя Tensorflow Probability MixtureNormal не работает, как в примере ⇐ Python

[Anonymous]

Версия Tensorflow — 2.17.1
Вероятностная версия Tensorflow — 0.24.0
Пример из документации https://www.tensorflow.org/probability/ ... rmal?hl=en следующий:
import numpy as np
import tensorflow as tf
import tensorflow_probability as tfp

tfd = tfp.distributions
tfpl = tfp.layers
tfk = tf.keras
tfkl = tf.keras.layers

# Load data -- graph of a [cardioid](https://en.wikipedia.org/wiki/Cardioid).
n = 2000
t = tfd.Uniform(low=-np.pi, high=np.pi).sample([n, 1])
r = 2 * (1 - tf.cos(t))
x = r * tf.sin(t) + tfd.Normal(loc=0., scale=0.1).sample([n, 1])
y = r * tf.cos(t) + tfd.Normal(loc=0., scale=0.1).sample([n, 1])

# Model the distribution of y given x with a Mixture Density Network.
event_shape = [1]
num_components = 5
params_size = tfpl.MixtureNormal.params_size(num_components, event_shape)
model = tfk.Sequential([
tfkl.Dense(12, activation='relu'),
tfkl.Dense(params_size, activation=None),
tfpl.MixtureNormal(num_components, event_shape)
])

# Fit.
batch_size = 100
model.compile(optimizer=tf.train.AdamOptimizer(learning_rate=0.02),
loss=lambda y, model: -model.log_prob(y))
model.fit(x, y,
batch_size=batch_size,
epochs=20,
steps_per_epoch=n // batch_size)

Это приводит к ошибке
ValueError: Only instances of `keras.Layer` can be added to a Sequential model. Received: (of type )


Подробнее здесь: https://stackoverflow.com/questions/793 ... in-example