def encodeData(model: Model, datax: tf.Tensor, datay: tf.Tensor) -> None: _, _, z = model.encodeData(datax, batchSize=64) plt.figure(figsize=(6, 6)) plt.scatter(z[:, 0], z[:, 1], c=datay, cmap='viridis') plt.colorbar() plt.show()