เทนเซอร์โฟลว์ :: ops :: InTopK

#include <nn_ops.h>

บอกว่าเป้าหมายอยู่ในการคาดการณ์ K อันดับต้น ๆ หรือไม่

สรุป

สิ่งนี้จะส่งออกอาร์เรย์บูล batch_size รายการ 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)$$

อาร์กิวเมนต์:

  • ขอบเขต: วัตถุ ขอบเขต
  • การคาดการณ์: เทนเซอร์ classes batch_size x
  • เป้าหมาย: เวกเตอร์ batch_size ของรหัสคลาส
  • k: จำนวนองค์ประกอบด้านบนที่ต้องพิจารณาเพื่อความแม่นยำในการประมวลผล

ผลตอบแทน:

  • 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 :: เอาท์พุท

 operator::tensorflow::Output() const