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

- training params are (again) attrubute of Entity

Browse Source 
- ditched layout concept
This commit is contained in:
jens
2026-01-02 14:51:19 +01:00
parent 8b97350564
commit 10691894d1
8 changed files with 70 additions and 98 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/rbm/stack_deep.py
+2 -2
View File
@@ -10,8 +10,8 @@ class StackDeep(Stack):
def train(self, batch: Mat, status=Status()):
_batch = np.copy(batch)
for index, layer in enumerate(self.layers):
print(f"Train layer {index} for {layer.training_params.num_epochs} epochs")
train(layer.entity, _batch, layer.training_params, status=status)
print(f"Train layer {index} for {layer.entity.training_params.num_epochs} epochs")
train(layer.entity, _batch, status=status)
_batch = layer.entity.forward(_batch)
def pass_up(self, visible: Mat, from_layer_id: int = 0):