spacecutter:PyTorch中的序数回归模型

太空切割机 spacecutter是一个用于在PyTorch中实现序数回归模型的库。 该库由模型和损失函数组成。 建议使用打包模型以使其与scikit-learn兼容。 安装 pip install spacecutter 用法 楷模 定义要生成单个标量预测值的任何PyTorch模型。 这将是我们的predictor模型。 然后，可以使用spacecutter.models.OrdinalLogisticModel封装该模型，该模型会将predictor的输出从单个数字转换为序数类概率的数组。 以下示例显示了针对具有三个序数类的问题的两层神经网络predictor如何执行此操作。 import numpy as np import torch from torch import nn from spacecutter . models import OrdinalLogisticMode