/** * @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_(predictions: T | TensorLike, targets: U | TensorLike, k?: number): Promise; export declare const inTopKAsync: typeof inTopKAsync_; export {};