0

请问你能帮我吗?我有适合不对称 sigmoid 函数的非线性数据。所以我生成了一个 sigmoid 函数。我已经在 curve_fit 函数中使用它来生成 beta1 和 beta2。但是我拥有的 sigmoid 函数只接受一个输入变量。是否有接受多个输入变量的 sigmoid 函数?这是我的 train_test_split、sigmoid 和 curve_fit 函数的代码:

X_train, X_test, y_train, y_test = train_test_split(df.drop('Montant TLPE', axis=1).values, df['Montant TLPE'].values, random_state=0, test_size=0.2)

def sigmoid(x, Beta_1, Beta_2):
    y = 1 / (1 + np.exp(-Beta_1*(x-Beta_2)))
    return y

from scipy.optimize import curve_fit
popt, pcov = curve_fit(sigmoid, np.array(X_train), np.array(y_train))
print(" beta_1 = %f, beta_2 = %f" % (popt[0], popt[1]))

非常感谢你的帮助。

4

1 回答 1

0

这是我的数据的散点图

于 2019-03-06T16:52:49.723 回答