def METHOD_NAME(self):
    self.model = HeteroNNHostModel(self.hetero_nn_param, self.flowid)
    self.model.set_transfer_variable(self.transfer_variable)
    self.model.set_partition(self.default_table_partitions)