Auto数据集回归预测

import pandas as pd
import statsmodels.formula.api as smf
import numpy as np
auto = pd.read_csv('https://liangdaima.com/static/data/statistics/Auto.csv')
auto['horsepower'] = pd.to_numeric(auto['horsepower'], errors='coerce')
auto = auto.dropna()
model = smf.ols('mpg ~ horsepower', data=auto).fit()
pred = model.get_prediction({'horsepower': 98})
print('预测值:', pred.predicted_mean)
print('95%置信区间:', pred.conf_int())