tensorflow::ops::InTopK

#include <nn_ops.h>

Menunjukkan apakah target tersebut termasuk dalam K prediksi teratas.

Ringkasan

Ini menghasilkan array boolean batch_size , sebuah entri out[i] bernilai true jika prediksi untuk kelas target termasuk di antara k prediksi teratas di antara semua prediksi, misalnya i . Perhatikan bahwa perilaku InTopK berbeda dari operasi TopK dalam penanganannya terhadap nilai yang sama; jika beberapa kelas memiliki nilai prediksi yang sama dan berada di sekitar batas k teratas, semua kelas tersebut dianggap berada di k teratas.

Secara lebih formal, mari kita

\(predictions_i\) misalnya prediksi untuk semua kelas i , \(targets_i\) misalnya kelas target i , \(out_i\) jadikan outputnya misalnya i ,

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

Argumen:

  • Cakupan: Objek Cakupan
  • prediksi: Tensor batch_size x classes .
  • Target: Vektor batch_size berisi ID kelas.
  • k: Jumlah elemen teratas yang akan diperiksa untuk menghitung presisi.

Mengembalikan:

  • Output : Presisi yang dihitung pada k sebagai bool Tensor .

Konstruktor dan Destruktor

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

Atribut publik

operation
precision

Fungsi publik

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

Atribut publik

operasi 

Operation operation

presisi 

::tensorflow::Output precision

Fungsi publik

InTopK 

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

simpul 

::tensorflow::Node * node() const

operator::tensorflow::Input 

 operator::tensorflow::Input() const

operator::tensorflow::Output 

 operator::tensorflow::Output() const