Tensorflow - 应用功能在1D张量

Question

我有一个函数dice

def dice(yPred,yTruth,thresh): 
    smooth = tf.constant(1.0) 
    threshold = tf.constant(thresh) 
    yPredThresh = tf.to_float(tf.greater_equal(yPred,threshold)) 
    mul = tf.mul(yPredThresh,yTruth) 
    intersection = 2*tf.reduce_sum(mul) + smooth 
    union = tf.reduce_sum(yPredThresh) + tf.reduce_sum(yTruth) + smooth 
    dice = intersection/union 
    return dice, yPredThresh 

其中工程。一个例子是这里

with tf.Session() as sess: 

    thresh = 0.5 
    print("Dice example") 
    yPred = tf.constant([0.1,0.9,0.7,0.3,0.1,0.1,0.9,0.9,0.1],shape=[3,3]) 
    yTruth = tf.constant([0.0,1.0,1.0,0.0,0.0,0.0,1.0,1.0,1.0],shape=[3,3]) 
    diceScore, yPredThresh= dice(yPred=yPred,yTruth=yTruth,thresh= thresh) 

    diceScore_ , yPredThresh_ , yPred_, yTruth_ = sess.run([diceScore,yPredThresh,yPred, yTruth]) 
    print("\nScore = {0}".format(diceScore_)) 

>>> Score = 0.899999976158

我希望能够遍历骰子的第三arguement给出脱粒。我不知道这样做的最佳方式，以便我可以从图中提取它。因为我不能环路成，显然分割一个TF张量大致如下的东西线...

def diceROC(yPred,yTruth,thresholds=np.linspace(0.1,0.9,20)): 
    thresholds = thresholds.astype(np.float32) 
    nThreshs = thresholds.size 
    diceScores = tf.zeros(shape=nThreshs) 

    for i in xrange(nThreshs): 
     score,_ = dice(yPred,yTruth,thresholds[i]) 
     diceScores[i] = score 
    return diceScores 

评估diceScoreROC产生错误'Tensor' object does not support item assignment。

Answer 1

而不是循环，我会鼓励你使用tensorflow的广播能力。如果重新定义dice到：

def dice(yPred,yTruth,thresh): 
    smooth = tf.constant(1.0) 
    yPredThresh = tf.to_float(tf.greater_equal(yPred,thresh)) 
    mul = tf.mul(yPredThresh,yTruth) 
    intersection = 2*tf.reduce_sum(mul, [0, 1]) + smooth 
    union = tf.reduce_sum(yPredThresh, [0, 1]) + tf.reduce_sum(yTruth, [0, 1]) + smooth 
    dice = intersection/union 
    return dice, yPredThresh 

您可以通过3维yPred和yTruth（假设张量将只是沿着最后一维重复）和1维thresh：

with tf.Session() as sess: 

    thresh = [0.1,0.9,20, 0.5] 
    print("Dice example") 
    yPred = tf.constant([0.1,0.9,0.7,0.3,0.1,0.1,0.9,0.9,0.1],shape=[3,3,1]) 
    ypred_tiled = tf.tile(yPred, [1,1,4]) 
    yTruth = tf.constant([0.0,1.0,1.0,0.0,0.0,0.0,1.0,1.0,1.0],shape=[3,3,1]) 
    ytruth_tiled = tf.tile(yTruth, [1,1,4]) 
    diceScore, yPredThresh= dice(yPred=ypred_tiled,yTruth=ytruth_tiled,thresh= thresh) 

    diceScore_ = sess.run(diceScore) 
    print("\nScore = {0}".format(diceScore_)) 

你会得到：

Score = [ 0.73333335 0.77777779 0.16666667 0.89999998]