transformer.ts

Typescript wrapper for @xenova/transformers.

Works in both commonjs and ESM module projects.

Supported Tasks

For all task, the model name can be optionally specified in the input object.

More tasks and models will be typed over time.

text-classification (alias: sentiment-analysis)

Input Type: { text: string }

Output Type: Array<{ label: TextClassificationLabel, score: number }>

1type TextClassificationLabel =
2  // Xenova/distilbert-base-uncased-finetuned-sst-2-english
3  | 'POSITIVE'
4  | 'NEGATIVE'
5  // Xenova/bert-base-multilingual-uncased-sentiment
6  | '5 stars'
7  | '4 stars'
8  | '3 stars'
9  | '2 stars'
10  | '1 star'
11  // Xenova/toxic-bert
12  | 'toxic'
13  | 'insult'
14  | 'obscene'
15  | 'identity_hate'
16  | 'threat'
17  | 'severe_toxic'

zero-shot-classification

Input Type: { text: string, labels: string[] }

Output Type: Array<{ label: string, score: number }>

License

This project is licensed with BSD-2-Clause

This is free, libre, and open-source software. It comes down to four essential freedoms [ref]:

• The freedom to run the program as you wish, for any purpose
• The freedom to study how the program works, and change it so it does your computing as you wish
• The freedom to redistribute copies so you can help others
• The freedom to distribute copies of your modified versions to others