/**
|
* @license
|
* Copyright 2019 Google LLC. All Rights Reserved.
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
* you may not use this file except in compliance with the License.
|
* You may obtain a copy of the License at
|
*
|
* http://www.apache.org/licenses/LICENSE-2.0
|
*
|
* Unless required by applicable law or agreed to in writing, software
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
* See the License for the specific language governing permissions and
|
* limitations under the License.
|
* =============================================================================
|
*/
|
import { Tensor } from '../tensor';
|
import { TensorLike } from '../types';
|
/**
|
* Returns whether the targets are in the top K predictions.
|
*
|
* ```js
|
* const predictions = tf.tensor2d([[20, 10, 40, 30], [30, 50, -20, 10]]);
|
* const targets = tf.tensor1d([2, 0]);
|
* const precision = await tf.inTopKAsync(predictions, targets);
|
* precision.print();
|
* ```
|
* @param predictions 2-D or higher `tf.Tensor` with last dimension being
|
* at least `k`.
|
* @param targets 1-D or higher `tf.Tensor`.
|
* @param k Optional Number of top elements to look at for computing precision,
|
* default to 1.
|
*/
|
/** @doc {heading: 'Operations', subheading: 'Evaluation'} */
|
declare function inTopKAsync_<T extends Tensor, U extends Tensor>(predictions: T | TensorLike, targets: U | TensorLike, k?: number): Promise<U>;
|
export declare const inTopKAsync: typeof inTopKAsync_;
|
export {};
|
