不对称数据拟合功能

Question

你能帮帮我吗？我有非线性数据，适合不对称的S形函数。所以我生成了一个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]))

非常感谢你的帮助。

Answer 1

Here is the scatter plot of my data