pytorch实战（交通事故理赔审核）

原创

于 2024-06-10 10:18:08 发布 · 199 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #python #深度学习

二分类问题。

数据集

变量名解释
CaseId 案例编号，没有实际意义
Q1 理赔员现场勘察采集的信息，Q1代表第一个问题的信息。信息被编码成数字，数字的大小不代表真实的关系。
Qk 同上，Qk代表第k个问题的信息。一共36个问题。
Evaluation 表示最终审核结果。0表示授予理赔，1表示未通过理赔审核。在test.csv中，这是需要被预测的标签。
http://sofasofa.io/competition.php?id=2

代码

import torch
import torch.nn as nn
from torch.utils.data import TensorDataset, DataLoader
from sklearn.model_selection import KFold
import pandas as pd


class my_module(nn.Module):
    def __init__(self, input_dim=36):
        super(my_module, self).__init__()
        self.fc1 = nn.Linear(input_dim, 16)
        self.relu1 = nn.ReLU()
        self.fc2 = nn.Linear(16, 1)
        self.sigmoid = nn.Sigmoid()

    def forward(self, x)