Home Reference Source Test
public class | source

UniversalEmbedding

You can directly use an instance of this class. universalEmbedding

Extends:

Tensor → UniversalEmbedding

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);        

Test:

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

This method return the core tensor instance

public

T: *

public

isTensor(object: Object ): Boolean

Check if object is tensor

Public Constructors

public constructor() source

Override:

Tensor#constructor

Public Members

public R: * source

public f: * source

public vecSize: number source

Private Members

private model: * source

Public Methods

public async connect(link: *): * source

Params:

NameTypeAttributeDescription
link *

Return:

*

public async sentenceEncode(sentences: *, asTensor: boolean): * source

Params:

NameTypeAttributeDescription
sentences *
asTensor boolean
  • optional
  • default: true

Return:

*

public async transform(tokenIdxs: *): * source

Params:

NameTypeAttributeDescription
tokenIdxs *

Return:

*