Star

🚀 著者表現のスタイルトランスフォーマー - STAR

このリポジトリは、著者表現のスタイルトランスフォーマー (STAR) モデルに関するものです。ここでは当該モデルの重みを公開しています。

再現性のために、STARのGitHubリポジトリ もご覧ください。

🚀 クイックスタート

STARモデルを使用して、文章の特徴抽出を行うことができます。以下に具体的なコード例を示します。

💻 使用例

基本的な使用法

tokenizer = AutoTokenizer.from_pretrained('roberta-large')
model = AutoModel.from_pretrained('AIDA-UPM/star')

examples = ['My text 1', 'This is another text']

def extract_embeddings(texts):
  encoded_texts = tokenizer(texts)
  with torch.no_grad():
    style_embeddings = model(encoded_texts.input_ids,
                             attention_mask=encoded_texts.attention_mask).pooler_output
  return style_embeddings

print(extract_embeddings(examples))

📄 ライセンス

論文を引用する場合は、以下のBibTeXエントリを使用してください。

@article{Huertas-Tato2023Oct,
	author = {Huertas-Tato, Javier and Martin, Alejandro and Camacho, David},
	title = {{Understanding writing style in social media with a supervised contrastively pre-trained transformer}},
	journal = {arXiv},
	year = {2023},
	month = oct,
	eprint = {2310.11081},
	doi = {10.48550/arXiv.2310.11081}
}