【发布时间】:2017-11-23 21:01:30
【问题描述】:
我正在使用 scipy.minimize 函数,我希望有一个参数只搜索带两位小数的选项。
def cost(parameters,input,target):
from sklearn.metrics import mean_squared_error
output = self.model(parameters = parameters,input = input)
cost = mean_squared_error(target.flatten(), output.flatten())
return cost
parameters = [1, 1] # initial parameters
res = minimize(fun=cost, x0=parameters,args=(input,target)
model_parameters = res.x
这里的 self.model 是一个基于参数执行一些矩阵操作的函数。输入和目标是两个矩阵。该函数按我想要的方式工作,除了我想让参数 [1] 有一个约束。理想情况下,我只想提供一个 numpy 数组,例如 np.arange(0,10,0.01)。这可能吗?
【问题讨论】: