【pytorch卷1】=1= 泰坦尼克---pytorch深度学习

Patrick的练习册

已于 2022-01-30 10:59:19 修改

阅读量614

点赞数

分类专栏： DL LJH-pytorch卷1 文章标签： pytorch 深度学习 python

于 2022-01-13 13:27:59 首次发布

本文链接：https://blog.csdn.net/Patrick6789/article/details/122471600

版权

搬运自LJH,已购买正版书，仅作为自己参考用。

DL的基本步骤，包括：

导入模组，data
特征工程
分成测试集训练集
组模型（损失函数，连接层,getloss)
训练模型，列出epoch。

################## 1. 导入 ###########################
#1.1 导入模组
import numpy as np

import torch
import torch.nn as nn
import torch.nn.functional as F
import os
from scipy import stats
import pandas as pd

#1.2 导入data
titanic_data = pd.read_csv("titanic3.csv")
print(titanic_data.columns )


################################## 2. 数据预处理 ######################

#2.1 dummies to one-hot
titanic_data = pd.concat([titanic_data,
                          pd.get_dummies(titanic_data['sex']),
                          pd.get_dummies(titanic_data['embarked'],prefix="embark"),
                          pd.get_dummies(titanic_data['pclass'],prefix="class")], axis=1)

print(titanic_data.columns )
print(titanic_data[