/usr/lib/python3/dist-packages/bumps/fitservice.py is in python3-bumps 0.7.6-3.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 | """
Fit job definition for the distributed job queue.
"""
from __future__ import print_function
import os
import sys
import json
import pickle
from . import cli
from . import __version__
# Site configuration determines what kind of mapper to use
# This should be true in cli.py as well
from .mapper import MPMapper as Mapper
from . import monitor
from .fitters import FitDriver
def fitservice(request):
import matplotlib
matplotlib.use('Agg')
path = os.getcwd()
service_version = __version__
request_version = str(request['version'])
if service_version != request_version:
raise ValueError('fitter version %s does not match request %s'
% (service_version, request_version))
data = request['data']
model = str(data['package'])
service_model_version = __version__
request_model_version = str(data['version'])
if service_model_version != request_model_version:
raise ValueError('%s version %s does not match request %s'
% (model, service_model_version, request_model_version))
options = pickle.loads(str(data['options']))
problem = pickle.loads(str(data['problem']))
problem.store = path
problem.output_path = os.path.join(path, 'model')
fitdriver = FitDriver(options.fit, problem=problem, **options)
fitdriver.mapper = Mapper.start_mapper(problem, options.args)
problem.show()
print("#", " ".join(sys.argv))
best, fbest = fitdriver.fit()
cli.save_best(fitdriver, problem, best)
matplotlib.pyplot.show()
return list(best), fbest
class ServiceMonitor(monitor.TimedUpdate):
"""
Display fit progress on the console
"""
def __init__(self, problem, path, progress=60, improvement=60):
monitor.TimedUpdate.__init__(self, progress=progress,
improvement=improvement)
self.path = path
self.problem = problem
self.images = []
def show_progress(self, history):
p = self.problem.getp()
try:
self.problem.setp(history.point[0])
dof = self.problem.dof
summary = self.problem.summarize()
finally:
self.problem.setp(p)
status = {
"step": history.step[0],
"cost": history.value[0] / dof,
"pars": history.point[0],
}
json_status = json.dumps(status)
open(os.path.join(self.path, 'status.json'), "wt").write(json_status)
status['table'] = summary
status['images'] = "\n".join('<img file="%s" alt="%s" />' % (f, f)
for f in self.images)
html_status = """\
<html><body>
Generation %(step)d, chisq %(cost)g
<pre>
%(table)s
</pre>
%(images)s
</body></html>
""" % status
open(os.path.join(self.path, 'status.html'), "wt").write(html_status)
def show_improvement(self, history):
import pylab
# print "step",history.step[0],"chisq",history.value[0]
self.problem.setp(history.point[0])
pylab.hold(False)
self.problem.plot(figfile=os.path.join(self.path, 'K'))
pylab.gcf().canvas.draw()
|