def __init__(self, fm_x, dim_x):
super(QMDensity, self).__init__()
self.fm_x = fm_x
self.dim_x = dim_x
self.qmd = layers.QMeasureDensity(dim_x)
self.cp = layers.CrossProduct()
self.num_samples = tf.Variable(initial_value=0., trainable=False)
def __init__(self, fm_x, dim_x, num_classes=2):
super(QMKDClassifier, self).__init__()
self.fm_x = fm_x
self.dim_x = dim_x
self.num_classes = num_classes
self.qmd = []
for _ in range(num_classes):
self.qmd.append(layers.QMeasureDensity(dim_x))
self.cp = layers.CrossProduct()
self.num_samples = tf.Variable(initial_value=tf.zeros((num_classes, )),
trainable=False)