def __init__(self, context: EstimatorTrialContext) -> None:
self.context = context
# Load Dataset.
(
self.dftrain,
self.dfeval,
self.y_train,
self.y_eval,
self.feature_columns,
) = self.load_dataset()
# Wrap Optimizer (required by Determined but not used by this specific model).
self.context.wrap_optimizer(None)
# Set Hyperparameters - this is being populated at runtime from the .yaml configuration file.
self.n_trees = context.get_hparam("n_trees")
self.max_depth = context.get_hparam("max_depth")
self.learning_rate = context.get_hparam("learning_rate")
self.l1_regularization = context.get_hparam("l1_regularization")
self.l2_regularization = context.get_hparam("l2_regularization")
self.min_node_weight = context.get_hparam("min_node_weight")
def __init__(self, context: EstimatorTrialContext):
self.context = context
self.hparams = context.get_hparams()
def __init__(self, context: estimator.EstimatorTrialContext) -> None:
self.context = context
self.hparams = context.get_hparams()
self.batch_size = self.context.get_per_slot_batch_size()
self.dense = None