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

[refactor] SubImage → SubImageExtract, add SubImageCombine

Browse Source 
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
jens
2026-06-02 08:20:38 +02:00
co-authored by Claude Sonnet 4.6
parent 2da2262368
commit a07698497d
5 changed files with 46 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mnist_test_sub_image.ipynb
+2 -2
View File
@@ -27,7 +27,7 @@
"from rbm.entity import Entity, EntityParams, TrainingParams\n",
"from rbm.matrix import Mat, np, rms_error_accu\n",
"from rbm.torch import Optimizer\n",
"from image.sub_image import SubImage\n",
"from image.sub_image import SubImageExtract\n",
"import math"
],
"outputs": [],
@@ -179,7 +179,7 @@
"source": [
"print(train_images.shape)\n",
"image_indices = np.random.randint(0, N-1, n_side*n_side)\n",
"sub_generator = SubImage(8,8,4,4)\n",
"sub_generator = SubImageExtract(8,8,4,4)\n",
"train_subimages = sub_generator(np.reshape(train_images, (N, 28,28)), 1)\n",
"print(train_subimages.shape)\n",
"train_subimages = np.reshape(train_subimages, (train_subimages.shape[0], 8*8))\n",