Star

🚀 用於作者身份表徵的風格變換器 - STAR

STAR（Style Transformer for Authorship Representations）是用於作者身份表徵的風格變換器模型。本倉庫提供了該模型的權重。你可以通過點擊論文鏈接查看更多詳細信息。

此外，你還可以訪問我們的 STAR GitHub 倉庫 進行模型復現。

🚀 快速開始

✨ 主要特性

本模型主要用於特徵提取，能夠從文本中提取風格嵌入向量，可應用於作者身份識別等相關領域。

📦 安裝指南

文檔未提供具體安裝步驟，可參考 STAR GitHub 倉庫 中的說明進行安裝。

💻 使用示例

基礎用法

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

📚 詳細文檔

文檔未提供詳細說明，可參考 STAR GitHub 倉庫 中的文檔。

📄 許可證

文檔未提供許可證信息。

📚 引用信息

如果你使用了本模型，請按照以下格式引用：

@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}
}