第一次推送代码
This commit is contained in:
commit
c15aa6b17a
|
@ -0,0 +1,27 @@
|
|||
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)
|
Loading…
Reference in New Issue