Commit ·
1741953
1
Parent(s): a791c56
Update app.py
Browse files
app.py
CHANGED
|
@@ -120,7 +120,7 @@ async def predict(request: TransactionRequest):
|
|
| 120 |
if request.use_rl:
|
| 121 |
ppo = load_ppo_model()
|
| 122 |
obs = pad[0]
|
| 123 |
-
action, _ = ppo.predict(obs)
|
| 124 |
print(f"RL Action: {action}, Observation: {obs}")
|
| 125 |
cat_idx = action // num_subcategories
|
| 126 |
subcat_idx = action % num_subcategories
|
|
|
|
| 120 |
if request.use_rl:
|
| 121 |
ppo = load_ppo_model()
|
| 122 |
obs = pad[0]
|
| 123 |
+
action, _ = ppo.predict(obs, deterministic=True) # Force deterministic
|
| 124 |
print(f"RL Action: {action}, Observation: {obs}")
|
| 125 |
cat_idx = action // num_subcategories
|
| 126 |
subcat_idx = action % num_subcategories
|