1
我有一个使用tensorflow函数的函数。我需要Theano的这个功能,因为在平台上我想使用这个代码只有Theano安装而不是tensorflow。我主要和Keras一起工作,所以tensorflow对我来说很神秘。 功能如下:是否可以将tensorflow代码转换为theano代码?
class WeightedBinaryCrossEntropy(object):
def __init__(self, pos_ratio):
neg_ratio = 1. - pos_ratio
self.pos_ratio = tf.constant(pos_ratio, tf.float32)
self.weights = tf.constant(neg_ratio/pos_ratio, tf.float32)
self.__name__ = "weighted_binary_crossentropy({0})".format(pos_ratio)
def __call__(self, y_true, y_pred):
return self.weighted_binary_crossentropy(y_true, y_pred)
def weighted_binary_crossentropy(self, y_true, y_pred):
# Transform to logits
epsilon = tf.convert_to_tensor(K.common._EPSILON, y_pred.dtype.base_dtype)
y_pred = tf.clip_by_value(y_pred, epsilon, 1 - epsilon)
y_pred = tf.log(y_pred/(1 - y_pred))
cost = tf.nn.weighted_cross_entropy_with_logits(y_true, y_pred, self.weights)
return K.mean(cost * self.pos_ratio, axis=-1)
model.compile(loss=WeightedBinaryCrossEntropy(0.05), optimizer=optimizer, metrics=['accuracy'])
的平台上安装Tensorflow是不可能的。 我从这里得到的代码https://github.com/fchollet/keras/issues/2115
那么Theano中的函数是否像Tensorflow中的函数一样工作?
谢谢你,这将需要一些时间来测试:测试后,我可能会接受它。 –