mirror of https://github.com/vladmandic/human
19 lines
648 B
TypeScript
19 lines
648 B
TypeScript
import * as tf from '../../dist/tfjs.esm.js';
|
|
declare type Tensor = typeof tf.Tensor;
|
|
declare type DB = Array<{
|
|
name: string;
|
|
source: string;
|
|
embedding: number[];
|
|
}>;
|
|
export declare function load(config: any): Promise<any>;
|
|
export declare function similarity(embedding1: any, embedding2: any, order?: number): number;
|
|
export declare function match(embedding: Array<number>, db: DB, threshold?: number): {
|
|
similarity: number;
|
|
name: string;
|
|
source: string;
|
|
embedding: number[];
|
|
};
|
|
export declare function enhance(input: any): Tensor;
|
|
export declare function predict(input: any, config: any): Promise<number[]>;
|
|
export {};
|