LordXido commited on
Commit
eaa19f0
·
verified ·
1 Parent(s): 3c4f7e6

Create trainer.py

Browse files
Files changed (1) hide show
  1. trainer.py +16 -0
trainer.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import numpy as np
2
+
3
+
4
+ class SelfTrainer:
5
+ def __init__(self, system):
6
+ self.system = system
7
+
8
+ def objective(self):
9
+ return -np.linalg.norm(self.system.Xi)
10
+
11
+ def train_step(self):
12
+ Psi = np.random.randn(self.system.n)
13
+ self.system.step(Psi)
14
+
15
+ if self.objective() > -1:
16
+ self.system.Xi *= 0.98