diff --git a/crypto/brian/index-deep-new.py b/crypto/brian/index-deep-new.py
index 4404189..4ca8610 100644
--- a/crypto/brian/index-deep-new.py
+++ b/crypto/brian/index-deep-new.py
@@ -59,6 +59,7 @@ BEST_DIR = os.path.join("models", "best")
os.makedirs(LAST_DIR, exist_ok=True)
os.makedirs(BEST_DIR, exist_ok=True)
CACHE_FILE = "candles_cache.json"
+TRAINING_CACHE_FILE = "training_cache.json"
# --- Constants ---
NUM_TIMEFRAMES = 6 # e.g., ["1s", "1m", "5m", "15m", "1h", "1d"]
@@ -66,6 +67,25 @@ NUM_INDICATORS = 20 # e.g., 20 technical indicators
FEATURES_PER_CHANNEL = 7 # Each channel has 7 features.
ORDER_CHANNELS = 1 # One extra channel for order information.
+# --- Training Cache Helpers ---
+def load_training_cache(filename):
+ if os.path.exists(filename):
+ try:
+ with open(filename, "r") as f:
+ cache = json.load(f)
+ print(f"Loaded training cache from {filename}.")
+ return cache
+ except Exception as e:
+ print("Error loading training cache:", e)
+ return {"total_pnl": 0.0}
+
+def save_training_cache(filename, cache):
+ try:
+ with open(filename, "w") as f:
+ json.dump(cache, f)
+ except Exception as e:
+ print("Error saving training cache:", e)
+
# --- Positional Encoding Module ---
class PositionalEncoding(nn.Module):
def __init__(self, d_model, dropout=0.1, max_len=5000):
@@ -376,7 +396,7 @@ def simulate_trades(model, env, device, args):
break
# --- Live HTML Chart Update (with Volume and Loss) ---
-def update_live_html(candles, trade_history, epoch, loss):
+def update_live_html(candles, trade_history, epoch, loss, total_pnl):
"""
Generate an HTML page with a live chart.
The chart displays price (line) and volume (bars on a secondary y-axis),
@@ -389,7 +409,7 @@ def update_live_html(candles, trade_history, epoch, loss):
fig, ax = plt.subplots(figsize=(12, 6))
update_live_chart(ax, candles, trade_history)
epoch_pnl = sum(trade["pnl"] for trade in trade_history)
- ax.set_title(f"Epoch {epoch} | Loss: {loss:.4f} | PnL: {epoch_pnl:.2f}| Total PnL: {epoch_pnl:.2f}")
+ ax.set_title(f"Epoch {epoch} | Loss: {loss:.4f} | PnL: {epoch_pnl:.2f}| Total PnL: {total_pnl:.2f}")
buf = BytesIO()
fig.savefig(buf, format='png')
plt.close(fig)
@@ -422,7 +442,7 @@ def update_live_html(candles, trade_history, epoch, loss):
-
Epoch {epoch} | Loss: {loss:.4f} | Total PnL: {epoch_pnl:.2f}
+
Epoch {epoch} | Loss: {loss:.4f} | PnL: {epoch_pnl:.2f}| Total PnL: {total_pnl:.2f}
