python实现多标签分类_python – nolearn用于多标签分类

正如Francisco Vargas所提到的,nolearn.dbn已被弃用,您应该使用nolearn.lasagne(如果可以的话).

如果要在千层面中进行多标签分类,则应将回归参数设置为True,定义验证分数和自定义丢失.

这是一个例子：

import numpy as np

import theano.tensor as T

from lasagne import layers

from lasagne.updates import nesterov_momentum

from nolearn.lasagne import NeuralNet

from nolearn.lasagne import BatchIterator

from lasagne import nonlinearities

# custom loss: multi label cross entropy

def multilabel_objective(predictions, targets):

epsilon = np.float32(1.0e-6)

one = np.float32(1.0)

pred = T.clip(predictions, epsilon, one - epsilon)

return -T.sum(targets * T.log(pred) + (one - targets) * T.log(one - pred), axis=1)

net = NeuralNet(

# customize "layers" to represent the architecture you want

# here I took a dummy architecture

layers=[(l