class_docstr: Abstract base class for UCB-style CB models.
class_docstr_tok: ["Abstract","base","class","for","UCB-style","CB","models","."]
code: def __init__(self, input_dim: int):
      super().__init__()
      self.input_dim = input_dim
  
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 1 to create a new UCBBaseModel object from inherited base classes: torch.nn, ABC with example: obj = UCBBaseModel(input_dim)