标题:Python 在希冀平台实现糖尿病数据集 BMI 一元线性回归模型,输出图像与期望不匹配
问题描述:我在使用 Scikit-Learn 的糖尿病数据集,选取 BMI 特征构建一元线性回归模型,需要在希冀平台上运行并输出符合要求的散点 + 回归线图像,但始终出现 “输出错误”,图像与期望样式不匹配(相似度 0%)。
需求细节:
数据集:sklearn.datasets.load_diabetes (),仅用 BMI 特征(第 3 列,索引 2)
模型:一元线性回归(LinearRegression)
输出要求:
散点:红色、无边缘色,标签 “BMI & Diabetes”
回归线:蓝色、实线,标签 “Regression Line”
坐标轴标签:x 轴 “BMI”,y 轴 “Diabetes Progression”
标题:“BMI & Diabetes”
运行环境:希冀平台(需强制保存图像到指定路径,平台通过图像相似度评分)
我的代码片段是
import numpy as np
import matplotlib.pyplot as plt
from sklearn.datasets import load_diabetes
from sklearn.linear_model import LinearRegression
# 1. 加载数据(全量数据训练,匹配期望的散点分布)
diabetes = load_diabetes()
X = diabetes.data[:, 2].reshape(-1, 1) # BMI特征
y = diabetes.target
# 2. 训练模型
model = LinearRegression()
model.fit(X, y)
y_pred = model.predict(X)
# 3. 绘制图像(严格匹配期望样式)
plt.figure(figsize=(6, 5)) # 画布大小匹配期望
# 散点:红色(无边缘色)
plt.scatter(X, y, color='red', edgecolors='none', label='BMI & Diabetes')
# 回归线:蓝色(实线,线宽2)
plt.plot(X, y_pred, color='blue', linewidth=2, label='Regression Line')
# 标签/标题:英文,与期望一致
plt.xlabel('BMI')
plt.ylabel('Diabetes Progression')
plt.title('BMI & Diabetes')
plt.legend(loc='upper left') # 图例位置匹配期望
# 4. 强制保存到平台默认路径(确保图像被识别)
plt.savefig('output.png', format='png', dpi=100, bbox_inches='tight')
plt.close()
实际结果是没有任何输出,以下是有关截图:

