jens/pyRBM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

improved tests

Browse Source
This commit is contained in:
jens
2026-01-03 12:51:43 +01:00
parent 529252166a
commit e56f99535a
2 changed files with 18 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/tests/test_learn_encoded_labels.py
+15 -5
View File
@@ -59,12 +59,22 @@ if __name__ == "__main__":
model.save()
# Plot
fig, axes = plt.subplots(1, len(encoded), figsize=(12, 3))
cmap = 'Grays'
fig, axes = plt.subplots(3, len(encoded), figsize=(12, 3))
for index, inp in enumerate(encoded):
img = np.reshape(inp, (label_w, label_h))
axes[index].imshow(img)
axes[index].axis('off')
axes[index].set_title(f'{test_labels[index]}')
hidden = model.forward(inp)
recon = model.backward(hidden)
img_hidden = np.reshape(hidden, (4, 4))
img_recon = np.reshape(recon, (label_w, label_h))
axes[0][index].imshow(img_hidden, cmap=cmap)
axes[0][index].axis('off')
axes[0][index].set_title(f'{test_labels[index]}')
axes[1][index].imshow(img_recon, cmap=cmap)
axes[1][index].axis('off')
axes[1][index].set_title(f'{test_labels[index]}')
axes[2][index].imshow(np.reshape(inp, (label_w, label_h)), cmap=cmap)
axes[2][index].axis('off')
axes[2][index].set_title(f'{test_labels[index]}')
plt.show()