2021-03-18 01:23:12 +01:00
|
|
|
import * as tf from '../../dist/tfjs.esm.js';
|
|
|
|
declare type Tensor = typeof tf.Tensor;
|
2021-03-18 01:16:40 +01:00
|
|
|
declare type DB = Array<{
|
|
|
|
name: string;
|
|
|
|
source: string;
|
|
|
|
embedding: number[];
|
|
|
|
}>;
|
2021-03-17 23:57:00 +01:00
|
|
|
export declare function load(config: any): Promise<any>;
|
2021-03-21 19:18:51 +01:00
|
|
|
export declare function similarity(embedding1: any, embedding2: any, order?: number): number;
|
2021-03-18 01:16:40 +01:00
|
|
|
export declare function match(embedding: Array<number>, db: DB, threshold?: number): {
|
2021-03-21 19:18:51 +01:00
|
|
|
similarity: number;
|
2021-03-17 23:57:00 +01:00
|
|
|
name: string;
|
|
|
|
source: string;
|
2021-03-18 01:16:40 +01:00
|
|
|
embedding: number[];
|
2021-03-17 23:57:00 +01:00
|
|
|
};
|
|
|
|
export declare function enhance(input: any): Tensor;
|
|
|
|
export declare function predict(input: any, config: any): Promise<number[]>;
|
|
|
|
export {};
|