论文复现 Rank consistent ordinal regression for neural networks withapplication to age estimation

DeniuHe

于 2021-07-06 11:46:47 发布

阅读量454

点赞数

分类专栏： Pytorch 算法

本文链接：https://blog.youkuaiyun.com/DeniuHe/article/details/118517439

版权

本文探讨了一种用于神经网络的秩一致有序回归方法，并将其应用于年龄估计问题。通过这种方法，可以提高预测模型对有序类别数据的准确性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import torch
import torch.nn.functional as F
from torch import nn
from torch.autograd import Variable
import pandas as pd
import numpy as np
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
from sklearn.preprocessing import Normalizer
from time import time

class RCORNN(torch.nn.Module):
    def __init__(self,nDim, n_hidden, nClass):
        self.nClass = nClass
        super(RCORNN, self).__init__()
        self.hidden_layer = torch.nn.Linear(nDim, n_hidden)
        self.output_layer = torch.nn.Linear(n_hidden, 1)
        self.linear_1_bias = torch.nn.Parameter(torch.ones(nClass-1).float())

    def forward(self, X):
        X = F.relu(self.hidden_layer(X))
        out = self.output_layer(X)
        logits = out + self.linear_1_bias