Views
No views yet
Files
pytorch_model.safetensors— model weightsconfig.json— model configurationREADME.md(this file)
lex) — focuses on token/phrase-level and domain terms common in 10-K and financial news.time) — encourages stability/consistency of relations across reporting periods and evolving events.lex, time) are also supported for analysis.⚠️ Not a generative LLM. Use it as an encoder (feature extractor).
1import torch
2from transformers import AutoTokenizer, AutoModel
3
4MODEL_ID = "william0816/Dual_View_Financial_Encoder"
5
6tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, use_fast=True)
7model = AutoModel.from_pretrained(MODEL_ID)
8
9def mean_pool(last_hidden_state, attention_mask):
10 # Mean-pool w.r.t. the attention mask
11 mask = attention_mask.unsqueeze(-1).type_as(last_hidden_state)
12 summed = (last_hidden_state * mask).sum(dim=1)
13 counts = torch.clamp(mask.sum(dim=1), min=1e-9)
14 return summed / counts
15
16texts = [
17 "The company faces supplier concentration risk due to a single-source vendor.",
18 "Management reported foreign exchange exposure impacting Q4 margins."
19]
20
21enc = tokenizer(texts, padding=True, truncation=True, return_tensors="pt")
22with torch.no_grad():
23 outputs = model(**enc)
24embeddings = mean_pool(outputs.last_hidden_state, enc["attention_mask"])
25
26# Cosine similarity for retrieval
27emb_norm = torch.nn.functional.normalize(embeddings, p=2, dim=1)
28similarity = emb_norm @ emb_norm.T
29print(similarity)1@misc{financial_risk_dualview_2025,
2 title = {Financial Risk Identification through Dual-view Adaptation},
3 author = {Chiu, Wei-Ning and collaborators},
4 year = {2025},
5 note = {Preprint/Project},
6 howpublished = {\url{https://huggingface.co/william0816/Dual_View_Financial_Encoder}}
7}