Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
caffe2.python.crf.CRFWithLoss Class Reference
Inheritance diagram for caffe2.python.crf.CRFWithLoss:

Public Member Functions

def __init__ (self, model, num_classes, transitions_blob=None)
def crf_loss (self, predictions, labels, seq_lengths=None)
def build_crf_net (self, input_blob, initial_state, transitions)
def update_predictions (self, classes)

Public Attributes


Detailed Description

Definition at line 17 of file

Member Function Documentation

def caffe2.python.crf.CRFWithLoss.build_crf_net (   self,
Adds the crf_net recurrent operator to the model.

model: model_helper.ModelHelper object new operators would be added

input_blob: the input sequence in a format T x N x D
where T is sequence size, N - batch size and D - input dimention
##Only supports batch-size 1##

seq_lengths: blob containing sequence lengths (unused)

Definition at line 230 of file

The documentation for this class was generated from the following file: