Commit 76fcd4a4 authored by abster12's avatar abster12
Browse files

added documentation

parent 83c5e29c
Loading
Loading
Loading
Loading
+10 −1
Original line number Diff line number Diff line
@@ -4305,10 +4305,19 @@ class GraphCNN(Layer):
    return result


class Hingeloss():
class Hingeloss(Layer):

  def __init__(self, in_layers=None, weights, batch_size, labels, reg,
               **kwargs):
    """
      Inputs have dimension D, there are C classes, and we operate on minibatches
      of N examples.
      Inputs:
      Weigths: A numpy array of shape (D, C) containing weights.
      batch_size: A numpy array of shape (N, D) containing a minibatch of data.
      labels: A numpy array of shape (N,)
      reg: (float) regularization strength
    """

    self.weights = weights
    self.batch_size = batch_size