UniversalEmbedding
Extends:
This UniversalEmbedding provide methods for transform sentences into its single vector representation based on use
Example:
import { universalEmbedding } from 'causal-net.representation';
import { termLogger } from 'causal-net.log';
import { tokenizer } from 'causal-net.preprocessing';
(async ()=>{
const BaseModelServer = 'http://0.0.0.0:8080/models/';
termLogger.groupBegin('load model');
await tokenizer.connect(BaseModelServer + 'use/vocab.json');
await universalEmbedding.connect(BaseModelServer + '/use/tensorflowjs_model.json');
termLogger.log('load finish');
const asEncode = true;
let tokens = [tokenizer.tokenize('dog', asEncode),
tokenizer.tokenize('cat', asEncode)];
termLogger.log({tokens});
let sentVec = await universalEmbedding.sentenceEncode(tokens);
sentVec.print();
let score = await universalEmbedding.encodeMatching(tokens[0], tokens[1]);
score.print();
termLogger.groupEnd();
})().catch(console.err);
Constructor Summary
Public Constructor | ||
public |
|
Member Summary
Public Members | ||
public |
R: * |
|
public |
f: * |
|
public |
vecSize: number |
Private Members | ||
private |
model: * |
Method Summary
Public Methods | ||
public |
async connect(link: *): * |
|
public |
async sentenceEncode(sentences: *, asTensor: boolean): * |
|
public |
async transform(tokenIdxs: *): * |
Inherited Summary
From class Tensor | ||
public get |
CoreTensor: * This method return the core tensor instance |
|
public |
T: * |
|
public |
isTensor(object: Object ): Boolean Check if object is tensor |