29 lines
631 B
Python
29 lines
631 B
Python
import torch
|
|
|
|
# 张量
|
|
import os
|
|
|
|
os.makedirs(os.path.join('..', 'data'), exist_ok=True)
|
|
data_file = os.path.join('..', 'data', 'house_tiny.csv')
|
|
with open(data_file, 'w') as f:
|
|
f.write('NumRooms,Alley,Price\n') # 列名
|
|
f.write('NA,Pave,127500\n') # 每行表示一个数据样本
|
|
f.write('2,NA,106000\n')
|
|
f.write('4,NA,178100\n')
|
|
f.write('NA,NA,140000\n')
|
|
|
|
import pandas as pd
|
|
|
|
data = pd.read_csv(data_file)
|
|
|
|
|
|
inputs, outputs = data.iloc[:, 0:1], data.iloc[:, 2]
|
|
inputs = inputs.fillna(inputs.mean())
|
|
|
|
mid = data.iloc[:, 1:2]
|
|
|
|
mid = pd.get_dummies(mid, dummy_na=True)
|
|
|
|
print(mid)
|
|
|
|
# 修改提交测试推送 |