code: def __init__(self, class_num) -> None:
      super().__init__()
      self.class_num = class_num
  
imports: import numpy
  import pandas
  import torch
  from fate.arch.dataframe import DataFrame
  from scipy.special import expit
  from fate.ml.utils.predict_tools import BINARY, MULTI, REGRESSION
label_desc: Use image node_id 1 to create a new CELoss object from inherited base classes: Loss with example: obj = CELoss(class_num)