1 |
|
2 |
|
3 |
|
4 |
|
5 |
|
6 |
|
7 |
|
8 |
|
9 |
|
10 |
|
11 | import { Tensor } from '@tensorflow/tfjs-core';
|
12 | import { categoricalCrossentropy as categoricalCrossentropyLoss, cosineProximity, meanAbsoluteError, meanAbsolutePercentageError, meanSquaredError, sparseCategoricalCrossentropy as sparseCategoricalCrossentropyLoss } from './losses';
|
13 | import { LossOrMetricFn } from './types';
|
14 | export declare function binaryAccuracy(yTrue: Tensor, yPred: Tensor): Tensor;
|
15 | export declare function categoricalAccuracy(yTrue: Tensor, yPred: Tensor): Tensor;
|
16 | export declare function precision(yTrue: Tensor, yPred: Tensor): Tensor;
|
17 | export declare function recall(yTrue: Tensor, yPred: Tensor): Tensor;
|
18 | export declare function binaryCrossentropy(yTrue: Tensor, yPred: Tensor): Tensor;
|
19 | export declare function sparseCategoricalAccuracy(yTrue: Tensor, yPred: Tensor): Tensor;
|
20 | export declare function topKCategoricalAccuracy(yTrue: Tensor, yPred: Tensor): Tensor;
|
21 | export declare function sparseTopKCategoricalAccuracy(yTrue: Tensor, yPred: Tensor): Tensor;
|
22 | export declare const mse: typeof meanSquaredError;
|
23 | export declare const MSE: typeof meanSquaredError;
|
24 | export declare const mae: typeof meanAbsoluteError;
|
25 | export declare const MAE: typeof meanAbsoluteError;
|
26 | export declare const mape: typeof meanAbsolutePercentageError;
|
27 | export declare const MAPE: typeof meanAbsolutePercentageError;
|
28 | export declare const categoricalCrossentropy: typeof categoricalCrossentropyLoss;
|
29 | export declare const cosine: typeof cosineProximity;
|
30 | export declare const sparseCategoricalCrossentropy: typeof sparseCategoricalCrossentropyLoss;
|
31 | export declare const metricsMap: {
|
32 | [functionName: string]: LossOrMetricFn;
|
33 | };
|
34 | export declare function get(identifier: string | LossOrMetricFn): LossOrMetricFn;
|
35 |
|
36 |
|
37 |
|
38 |
|
39 |
|
40 |
|
41 |
|
42 |
|
43 |
|
44 |
|
45 |
|
46 |
|
47 |
|
48 |
|
49 |
|
50 |
|
51 |
|
52 | export declare function getLossOrMetricName(fn: string | LossOrMetricFn): string;
|