Change variable name for better understanding
hidden_dims here refers to the number of the output of first hidden layer
"metadata": {},
"outputs": [],
"source": [
- "num_output_dims = 2 if smoke_test else 10\n",
+ "num_hidden_dims = 2 if smoke_test else 10\n",
"\n",
"\n",
"class DeepGP(DeepGP):\n",
" def __init__(self, train_x_shape):\n",
" hidden_layer = ToyDeepGPHiddenLayer(\n",
" input_dims=train_x_shape[-1],\n",
- " output_dims=num_output_dims,\n",
+ " output_dims=num_hidden_dims,\n",
" mean_type='linear',\n",
" )\n",
" \n",
