import pandas as pd
import torch
train_data = pd.read_csv("./titanic/train.csv")
features = ["Sex", "SibSp", "Parch",'Pclass']
X = pd.get_dummies(train_data[features])
a=torch.FloatTensor(X.values)
dataFrame数据转化为张量
于 2022-03-03 16:59:09 首次发布