تينسورفلو :: العمليات :: InTopK

#include <nn_ops.h>

يقول ما إذا كانت الأهداف في أعلى تنبؤات K

ملخص

ينتج عن هذا مصفوفة batch_size bool ، يكون الإدخال out[i] true إذا كان التنبؤ للفئة المستهدفة من بين أعلى تنبؤات k بين جميع التنبؤات على سبيل المثال i . لاحظ أن سلوك InTopK يختلف عن TopK op في تعامله مع العلاقات ؛ إذا كانت فئات متعددة قيمة التنبؤ نفسها، وتنتشر على الأعلى- k الحدود، وتعتبر كل من تلك الفئات ليكون في الجزء العلوي k .

بشكل أكثر رسمية ، دعونا

\(predictions_i\) هي التنبؤات لجميع الفئات على سبيل المثال i ، \(targets_i\) تكون الفئة المستهدفة على سبيل المثال i ، \(out_i\) يكون الناتج على سبيل المثال i ،

$$out_i = predictions_{i, targets_i} TopKIncludingTies(predictions_i)$$

الحجج:

  • النطاق: كائن النطاق
  • التنبؤات: batch_size x classes موتر.
  • الأهداف: ناقل batch_size الفئة.
  • ك: عدد العناصر العليا التي يجب النظر إليها من أجل دقة الحوسبة.

عائدات:

  • Output : الدقة المحسوبة عند k bool Tensor .

البنائين والمدمرين

InTopK (const :: tensorflow::Scope & scope, :: tensorflow::Input predictions, :: tensorflow::Input targets, int64 k)

السمات العامة

operation
precision

الوظائف العامة

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

السمات العامة

عملية

Operation operation

الاحكام

::tensorflow::Output precision

الوظائف العامة

InTopK

 InTopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input predictions,
  ::tensorflow::Input targets,
  int64 k
)

العقدة

::tensorflow::Node * node() const 

المشغل :: tensorflow :: الإدخال

 operator::tensorflow::Input() const 

المشغل :: Tensorflow :: Output

 operator::tensorflow::Output() const