HONGRIZON commited on
Commit
9733c0c
·
verified ·
1 Parent(s): c6a8235

Delete configuration_jnu_tsb.py

Browse files
Files changed (1) hide show
  1. configuration_jnu_tsb.py +0 -82
configuration_jnu_tsb.py DELETED
@@ -1,82 +0,0 @@
1
- from __future__ import annotations
2
-
3
- from typing import Any, Dict, List, Optional
4
-
5
- from transformers import PretrainedConfig
6
-
7
-
8
- class JNUTSBConfig(PretrainedConfig):
9
- """Configuration for JNU-TSB.
10
-
11
- JNU-TSB stores a lightweight routing wrapper. It does not store Chronos-2
12
- or Polyglot-Ko weights; those base models are loaded by the runtime when
13
- they are needed.
14
- """
15
-
16
- model_type = "jnu_tsb"
17
-
18
- def __init__(
19
- self,
20
- repo_id: str = "HONGRIZON/JNU-TSB",
21
- display_name: str = "JNU-TSB",
22
- full_name: str = "Jeju National University Time-Series Bridge",
23
- nickname: str = "Time-Series Bridge / Time-Series Seungbin",
24
- chronos_model_id: str = "amazon/chronos-2",
25
- llm_model_id: str = "EleutherAI/polyglot-ko-1.3b",
26
- route_name: str = "jnu-tsb",
27
- quantile_levels: Optional[List[float]] = None,
28
- event_categories: Optional[List[str]] = None,
29
- covariate_columns: Optional[List[str]] = None,
30
- use_llm_extractor: bool = True,
31
- **kwargs: Any,
32
- ) -> None:
33
- super().__init__(**kwargs)
34
- self.repo_id = repo_id
35
- self.display_name = display_name
36
- self.full_name = full_name
37
- self.nickname = nickname
38
- self.chronos_model_id = chronos_model_id
39
- self.llm_model_id = llm_model_id
40
- self.route_name = route_name
41
- self.quantile_levels = quantile_levels or [0.1, 0.5, 0.9]
42
- self.event_categories = event_categories or [
43
- "earnings",
44
- "product",
45
- "macro",
46
- "regulation",
47
- "supply_chain",
48
- "competition",
49
- "other",
50
- ]
51
- self.covariate_columns = covariate_columns or [
52
- "cov_earnings_count",
53
- "cov_product_count",
54
- "cov_macro_count",
55
- "cov_regulation_count",
56
- "cov_supply_chain_count",
57
- "cov_competition_count",
58
- "cov_other_count",
59
- "cov_sentiment_pos_count",
60
- "cov_sentiment_neg_count",
61
- "cov_sentiment_neu_count",
62
- "cov_news_count",
63
- "cov_sentiment_mean",
64
- "cov_confidence_mean",
65
- "cov_event_score",
66
- ]
67
- self.use_llm_extractor = bool(use_llm_extractor)
68
-
69
- def to_runtime_dict(self) -> Dict[str, Any]:
70
- return {
71
- "repo_id": self.repo_id,
72
- "display_name": self.display_name,
73
- "full_name": self.full_name,
74
- "nickname": self.nickname,
75
- "chronos_model_id": self.chronos_model_id,
76
- "llm_model_id": self.llm_model_id,
77
- "route_name": self.route_name,
78
- "quantile_levels": self.quantile_levels,
79
- "event_categories": self.event_categories,
80
- "covariate_columns": self.covariate_columns,
81
- "use_llm_extractor": self.use_llm_extractor,
82
- }