哈哈
This commit is contained in:
Binary file not shown.
Binary file not shown.
@@ -211,5 +211,11 @@ def get_live_signal(period: int = None, model_type: str = 'lightgbm',
|
|||||||
return 0
|
return 0
|
||||||
X_scaled = scaler.transform(X_last)
|
X_scaled = scaler.transform(X_last)
|
||||||
X_scaled_df = pd.DataFrame(X_scaled, columns=feature_cols, index=X_last.index)
|
X_scaled_df = pd.DataFrame(X_scaled, columns=feature_cols, index=X_last.index)
|
||||||
pred = model.predict(X_scaled_df)
|
proba = model.predict_proba(X_scaled_df)[0] # (p0, p1, p2)
|
||||||
return int(pred[0])
|
pred = model.predict(X_scaled_df)[0]
|
||||||
|
confidence_threshold = MC.get('confidence_threshold', 0.45)
|
||||||
|
logger.info(f"方案B 预测概率: 观望={proba[0]:.2f} 做多={proba[1]:.2f} 做空={proba[2]:.2f} -> {int(pred)}")
|
||||||
|
if proba.max() < confidence_threshold:
|
||||||
|
logger.info(f"置信度 {proba.max():.2f} < {confidence_threshold},视为观望")
|
||||||
|
return 0
|
||||||
|
return int(pred)
|
||||||
|
|||||||
@@ -705,6 +705,7 @@ class BitmartFuturesTransaction:
|
|||||||
self.last_kline_time = current_15m_id
|
self.last_kline_time = current_15m_id
|
||||||
logger.info(f"进入新 15m K 线: {current_15m_id}")
|
logger.info(f"进入新 15m K 线: {current_15m_id}")
|
||||||
raw = get_live_signal(period=15)
|
raw = get_live_signal(period=15)
|
||||||
|
logger.info(f"方案B 信号: {raw} (0=观望 1=做多 2=做空), 当前持仓: {self.start}")
|
||||||
if raw == 1:
|
if raw == 1:
|
||||||
if self.start == 0:
|
if self.start == 0:
|
||||||
signal = ('long', current_price)
|
signal = ('long', current_price)
|
||||||
|
|||||||
Reference in New Issue
Block a user