1234567891011121314151617181920212223242526272829303132333435363738394041424344454647 |
- import scipy.optimize
- import math
- def himmelblau(x, y):
- return (x * x + y - 11) * ( x * x + y - 11) + (x + y * y - 7) * (x + y * y - 7)
- def beale(x, y):
- return math.pow(1.5 - x + x*y, 2) + math.pow(2.25 - x + x*y*y, 2) + math.pow(2.625 - x + x*y*y*y, 2);
- def main():
- if True:
- initial = [-3.670609291875735,3.8585484651848674]
- solution = scipy.optimize.fmin(lambda x: beale(x[0], x[1]),
- initial, retall=True)
- print "loss", beale(solution[0][0], solution[0][1])
- elif False:
- def banana(x, y):
- return (1 - x) * (1 - x) + 100 * (y - x * x) * ( y - x * x)
- initial = [-1.675793744623661,-1.945310341194272]
- solution = scipy.optimize.fmin(lambda x: banana(x[0], x[1]),
- initial, retall=True)
- elif False:
- initial = [4.474377192556858, 0.22207495383918285]
- initial = [-7.185110699385405, 0.01616438291966915]
- solution = scipy.optimize.fmin(lambda x: himmelblau(x[0], x[1]),
- initial, retall=True)
- else:
- solution = scipy.optimize.fmin(lambda x: (x[0]-10) * (x[0]-10), [0], retall=True)
- print solution[0]
- for i, s in enumerate(solution[1]):
- print str(i) + ":", s
- if __name__ == "__main__":
- main()
|