IMP logo
IMP Reference Guide  2.14.0
The Integrative Modeling Platform
kernel/write_an_optimizer_state.py

While we do not recommend doing serious work using optimizer states written in Python, it is often useful when prototyping or testing code. Copy this example and modify as needed.

1 ## \example kernel/write_an_optimizer_state.py
2 # While we do not recommend doing serious work using optimizer states
3 # written in Python, it is often useful when prototyping or testing code.
4 # Copy this example and modify as needed.
5 
6 from __future__ import print_function
7 import IMP
8 import sys
9 
10 IMP.setup_from_argv(sys.argv, "Optimizer state")
11 
12 
13 class MyOptimizerState(IMP.OptimizerState):
14  """An optimizer state which prints out the last scores of some restraints"""
15 
16  def __init__(self, rs):
17  IMP.OptimizerState.__init__(self, rs[0].get_model(),
18  "MyOptimizerState%1%")
19  self.rs = rs
20 
21  def update(self):
22  for r in self.rs:
23  print(r.get_name(), r.get_last_score())
24 
25 # some code to create and evaluate it
26 k = IMP.FloatKey("a key")
27 m = IMP.Model()
28 # we don't have any real restraints in the kernel
29 r0 = IMP._ConstRestraint(m, [], 1)
30 r0.set_name("restraint 0")
31 
32 r1 = IMP._ConstRestraint(m, [], 2)
33 r1.set_name("restraint 1")
34 
35 rs = IMP.RestraintSet([r0, r1], 1.0)
36 sf = rs.create_scoring_function()
37 
38 os = MyOptimizerState([r0, r1])
39 os.set_name("python optimizer state")
40 # we don't have any optimizers either
41 co = IMP._ConstOptimizer(m)
42 co.set_scoring_function(sf)
43 co.add_optimizer_state(os)
44 print(co.optimize(100))
45 
46 # this is needed to clean up memory properly for some reason
47 co.remove_optimizer_state(os)
48 del os
49 del m