From d86e29cef980bdf8ead0d479ced19a7b1ce8e222 Mon Sep 17 00:00:00 2001 From: Alex Nikulkov Date: Tue, 30 Jan 2024 22:08:40 -0800 Subject: [PATCH] ap_container Config changes Summary: 1. Add NE metric logging 2. Remove log label transform from defaults 3. Update data loading configs: table name, ds, batch size Reviewed By: BerenLuthien Differential Revision: D53208098 fbshipit-source-id: fcde1fb4296ebc30e28aef927d71d9a62f1fb59c --- .../policy_learners/contextual_bandits/neural_linear_bandit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pearl/policy_learners/contextual_bandits/neural_linear_bandit.py b/pearl/policy_learners/contextual_bandits/neural_linear_bandit.py index fc4e3e24..b47e941d 100644 --- a/pearl/policy_learners/contextual_bandits/neural_linear_bandit.py +++ b/pearl/policy_learners/contextual_bandits/neural_linear_bandit.py @@ -63,7 +63,7 @@ def __init__( learning_rate: float = 0.0003, l2_reg_lambda_linear: float = 1.0, state_features_only: bool = False, - loss_type: str = "mse", # one of the LOSS_TYPES names, e.g., mse, mae, xentropy + loss_type: str = "mse", # one of the LOSS_TYPES names: [mse, mae, cross_entropy] output_activation_name: str = "linear", use_batch_norm: bool = False, use_layer_norm: bool = False,