import pandas as pd
# 定义特征名称
# ['经度', '纬度', '住房年龄中位数', '房间总数', '卧室总数', '人口', '住户', '收入中位数', '房屋价值中位数']
column_names = ['longitude', 'latitude', 'housingMedianAge', 'totalRooms', 'totalBedrooms', 'population', 'households', 'medianIncome', 'medianHouseValue']
# 读取数据文件
data_path = r"E:\地理探测器\SHAP\CaliforniaHousing\cal_housing.data"  # 根据实际情况修改路径
data = pd.read_csv(data_path, sep=',', header=None, names=column_names)
# 分离特征矩阵X和目标向量Y
X = data.drop('medianHouseValue', axis=1)
y = data['medianHouseValue']

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐