Kaggle的一个入门级比赛Tatanic数据集二分类（python实现）

最新推荐文章于 2024-08-25 17:21:24 发布

lblbc

最新推荐文章于 2024-08-25 17:21:24 发布

阅读量751

点赞数

文章标签： python pytorch 机器学习神经网络

本文链接：https://blog.csdn.net/weixin_45517493/article/details/118533653

版权

该博客分享了一篇使用PyTorch实现的Kaggle入门级比赛——Titanic数据集的二分类问题。作者通过训练神经网络，达到了0.76的得分，并提供了数据链接和代码示例。

摘要由CSDN通过智能技术生成

这是一个kaggle入门级比赛。程序基于pytorch，目的是利用数据集train_data训练一个神经网络，用以预测test_data的结果。上传kaggle后最终得分0.76(我很菜)。数据链接如下

https://www.kaggle.com/c/titanic/data

代码如下：

import torch
from torch.utils.data import Dataset
import numpy as np 
import pandas as pd 
import matplotlib as mlp
class TatanicDataset(Dataset):
    def __init__(self,filepath):
    	# 从原始数据集中取五个特征
        features = ["Pclass", "Sex", "SibSp", "Parch", "Fare"]
        data = pd.read_csv(filepath)
        self.len = data.shape[0]  # shape(多少行，多少列)
        
        # data[features]的类型是DataFrame,进行onehot表示
        self.x_data = torch.from_numpy(np.array(pd.get_dummies(data[features])))