Stack: added state load
This commit is contained in:
@@ -47,12 +47,18 @@ def main():
|
||||
# Create stack from project file
|
||||
stack = StackFactory.from_file("/home/jens/work/repos/Rbm/test.prj")
|
||||
|
||||
# Load state
|
||||
stack.state_load()
|
||||
|
||||
# Load train data
|
||||
batch = read_armadillo("/home/jens/work/repos/Rbm/test.training.dat")
|
||||
|
||||
# Train
|
||||
stack.train(batch)
|
||||
|
||||
# Save state
|
||||
stack.state_save()
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
print("Test: [passed]")
|
||||
|
||||
Reference in New Issue
Block a user