Compare commits
2 Commits
efd9a08212
...
4a5397f153
Author | SHA1 | Date | |
---|---|---|---|
![]() |
4a5397f153 | ||
![]() |
65bdf3256e |
@@ -41,9 +41,12 @@ The subcommands available at this time are `gather` and `evaluate`.
|
|||||||
|
|
||||||
## Functions
|
## Functions
|
||||||
|
|
||||||
* ~recommend~ (in-progress)
|
* recommend
|
||||||
|
|
||||||
Infer a UCS category for a text description.
|
Infer a UCS category for a text description. Text metadata is extracted from
|
||||||
|
audio files and the language model can recommend a corresponding list of
|
||||||
|
appropriate categories, ranked by their alignment with the category
|
||||||
|
definition.
|
||||||
|
|
||||||
* gather
|
* gather
|
||||||
|
|
||||||
|
@@ -2,8 +2,6 @@ import os
|
|||||||
import sys
|
import sys
|
||||||
import csv
|
import csv
|
||||||
|
|
||||||
from typing import Optional
|
|
||||||
|
|
||||||
from sentence_transformers import SentenceTransformer
|
from sentence_transformers import SentenceTransformer
|
||||||
import tqdm
|
import tqdm
|
||||||
import click
|
import click
|
||||||
@@ -51,6 +49,11 @@ def recommend(text, paths, model):
|
|||||||
"""
|
"""
|
||||||
Infer a UCS category for a text description
|
Infer a UCS category for a text description
|
||||||
|
|
||||||
|
"Description" text metadata is extracted from audio files given as PATHS,
|
||||||
|
or text can be provided directly using the "--text" option. The selected
|
||||||
|
model is then used to attempt to classify the given text according to
|
||||||
|
the synonyms and explanations definied for each UCS subcategory. A list
|
||||||
|
of ranked subcategories is printed to the terminal for each PATH.
|
||||||
"""
|
"""
|
||||||
m = SentenceTransformer(model)
|
m = SentenceTransformer(model)
|
||||||
ctx = InferenceContext(m, model)
|
ctx = InferenceContext(m, model)
|
||||||
|
Reference in New Issue
Block a user