tensorflow::ops::InTopK

#include <nn_ops.h>

Cho biết liệu các mục tiêu có nằm trong top K dự đoán hay không.

Bản tóm tắt

Thao tác này xuất ra một mảng boolean batch_size , một mục out[i] là true nếu dự đoán cho lớp mục tiêu nằm trong số k dự đoán hàng đầu trong số tất cả các dự đoán, ví dụ i . Lưu ý rằng hành vi của InTopK khác với toán tử TopK trong việc xử lý các trường hợp trùng lặp; nếu nhiều lớp có cùng giá trị dự đoán và nằm ở ranh giới top- k , tất cả các lớp đó đều được coi là nằm trong top k .

Nói một cách trang trọng hơn, hãy để

\(predictions_i\) Ví dụ, hãy đưa ra dự đoán cho tất cả các lớp, chẳng hạn như i . \(targets_i\) Ví dụ, i là lớp mục tiêu. \(out_i\) Ví dụ, đầu ra là i .

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

Luận điểm:

  • phạm vi: Một đối tượng Phạm vi
  • Dự đoán: Một tensor batch_size x classes .
  • targets: Một vectơ batch_size chứa ID lớp.
  • k: Số lượng phần tử hàng đầu cần xem xét để tính toán độ chính xác.

Trả về:

  • Output : Độ chính xác được tính toán tại k dưới dạng bool Tensor .

Hàm khởi tạo và hàm hủy

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

Thuộc tính công khai

operation
precision

Chức năng công cộng

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

Thuộc tính công khai

hoạt động 

Operation operation

độ chính xác 

::tensorflow::Output precision

Chức năng công cộng

InTopK 

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

nút 

::tensorflow::Node * node() const

operator::tensorflow::Input 

 operator::tensorflow::Input() const

operator::tensorflow::Output 

 operator::tensorflow::Output() const