Skip to content

Commit

Permalink
Added model types.
Browse files Browse the repository at this point in the history
  • Loading branch information
mlucy-stripe committed Feb 13, 2019
1 parent 0216004 commit 46c8a76
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions basilica/basilica/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -247,6 +247,13 @@ def embed_sentences(self, sentences, model='english', version='default',
:param sentences: An iterable (such as a list) of sentences to embed.
:type sentences: Iterable[str]
:param model: What model to use (i.e. the kind of sentence being embedded).
* **generic:** Generic English text embedding (the default.)
* **reddit:** Text embedding specialized for English Reddit posts.
* **twitter:** Text embedding specialized for English tweets.
* **email:** Text embedding specialized for English emails.
* **product-reviews:** Text embedding specialized for English product reviews.
:type model: str
:param version: What version of that model to use.
:type version: str
Expand Down Expand Up @@ -284,6 +291,13 @@ def embed_sentence(self, sentence, model='english', version='default',
:param sentence: The sentence to embed.
:type sentence: str
:param model: What model to use (i.e. the kind of sentence being embedded).
* **generic:** Generic English text embedding (the default.)
* **reddit:** Text embedding specialized for English Reddit posts.
* **twitter:** Text embedding specialized for English tweets.
* **email:** Text embedding specialized for English emails.
* **product-reviews:** Text embedding specialized for English product reviews.
:type model: str
:param version: What version of that model to use.
:type version: str
Expand Down

0 comments on commit 46c8a76

Please sign in to comment.