如何預測使用statsmodels.formula.api（蟒蛇）

Question

我使用下面的訓練Logistic模型，從乳腺癌的數據，並僅使用一個功能「mean_area」

from statsmodels.formula.api import logit 
logistic_model = logit('target ~ mean_area',breast) 
result = logistic_model.fit() 

有一個內置的新值在訓練好的模型中預測方法。但是，它給出了所有訓練樣本的預測值。如下

predictions = result.predict() 

假設我想要一個新值的預測說30如何使用訓練的模型出來的價值？ （而不是手動讀取係數和計算）

Answer 1

您可以爲.predict()模型提供新值，如notebook中的輸出＃11所示，來自單個觀測的文檔。您可以提供多個觀察值爲2d array，例如DataFrame - see docs。

由於您使用的公式API，您的輸入必須在pd.DataFrame的形式，以便列引用是可用的。在你的情況下，你可以使用類似.predict(pd.DataFrame({'mean_area': [1,2,3]})。

statsmodels.predict()使用用於不提供替代時作爲默認僅擬合觀測。