human/types/embedding/embedding.d.ts

19 lines
648 B
TypeScript
Raw Normal View History

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 {};