@ -35,9 +35,8 @@ def train_with_parameters(
"""
model = constellation.ConstellationNet(
order=order,
encoder_layers_sizes=layer_sizes,
decoder_layers_sizes=layer_sizes[::-1],
channel_model=constellation.GaussianChannel()
encoder_layers=layer_sizes,
decoder_layers=layer_sizes[::-1],
)
# List of training examples (not shuffled)