fix: Update TradingBot signal processing to handle NaN values and improve MLFilter ONNX session configuration

2026-03-02 00:47:17 +09:00
parent 031adac977
commit 361b0f4e00
3 changed files with 15 additions and 5 deletions
--- a/src/ml_filter.py
+++ b/src/ml_filter.py
@@ -53,8 +53,12 @@ class MLFilter:
        if self._onnx_path.exists():
            try:
                import onnxruntime as ort
+                sess_opts = ort.SessionOptions()
+                sess_opts.intra_op_num_threads = 1
+                sess_opts.inter_op_num_threads = 1
                self._onnx_session = ort.InferenceSession(
                    str(self._onnx_path),
+                    sess_options=sess_opts,
                    providers=["CPUExecutionProvider"],
                )
                self._lgbm_model = None