class_docstr: Abstract base class for UCB-style CB models.
class_docstr_tok: ["Abstract","base","class","for","UCB-style","CB","models","."]
docstr: Model forward pass.
  Returns pred_label, pred_sigma, ucb (where ucb = pred_label + ucb_alpha*pred_sigma)
docstr_tok: ["Model","forward","pass",".","Returns","pred_label",",","pred_sigma",",","ucb","(","where","ucb","=","pred_label","+","ucb_alpha","*","pred_sigma",")"]
code: def forward(
      self, inp: torch.Tensor, ucb_alpha: Optional[float] = None
  ) -> Dict[str, torch.Tensor]:
      """
      Model forward pass.
      Returns pred_label, pred_sigma, ucb (where ucb = pred_label + ucb_alpha*pred_sigma)
      """
      pass
  
file_path: ReAgent/reagent/models/cb_base_model.py
imports: import logging
  from abc import ABC, abstractmethod
  from typing import Dict, Optional
  import torch
label_desc: Use image node_id 3 for calling the UCBBaseModel obj's underlying member method code with example usage: obj.forward(inp, ucb_alpha) without return types