날짜 수정
This commit is contained in:
@@ -56,6 +56,9 @@ class StockLSTM:
|
||||
self.dropout = dropout
|
||||
self.model = None
|
||||
self.device = None
|
||||
# 정규화 파라미터 — train_model() 또는 load() 전까지 None
|
||||
self.seq_mean: Optional[np.ndarray] = None
|
||||
self.seq_std: Optional[np.ndarray] = None
|
||||
self._build()
|
||||
|
||||
def _build(self):
|
||||
@@ -148,7 +151,7 @@ class StockLSTM:
|
||||
def predict_proba(self, X_seq: np.ndarray) -> np.ndarray:
|
||||
"""예측 확률 반환 (N, 3)"""
|
||||
torch, nn = _get_torch()
|
||||
if torch is None or self.model is None:
|
||||
if torch is None or self.model is None or self.seq_mean is None or self.seq_std is None:
|
||||
return np.full((len(X_seq), 3), 1.0 / 3)
|
||||
|
||||
self.model.eval()
|
||||
|
||||
Reference in New Issue
Block a user