| date | section | title |
|---|---|---|
2021年 九月 1日 星期三 |
3 |
galileo.framework.tf.python.keras_trainer.KerasTrainer |
galileo.framework.tf.python.keras_trainer.KerasTrainer - Trainer for tf keras
\
继承自 galileo.framework.tf.python.tf_trainer.TFTrainer .
def init_model (self, **kwargs)
init model
def get_dataset (self, mode)
get an dataset
def do_train (self)\
def do_evaluate (self)\
def do_predict (self)\
model\
optimizer\
latest_ckp\
Trainer for tf keras
注意
API: galileo.tf.KerasTrainer
get an dataset
参数
mode train/evaluate/predict
重载 galileo.framework.tf.python.tf_trainer.TFTrainer .
init model
注解
all args are stored in run_config
参数
model_dir model dir
inputs_fn inputs function, requried when self.inputs is None
batch_size Mini-batch size
max_id max vertex id
batch_num Number of mini-batch, default is [max_id] / [batch_size]
log_steps Number of steps to print log
log_max_times_per_epoch log max times per epoch, default is 100
start_epoch start of epoch
num_epochs number epochs
optimizer adam, sgd, momentum, adagrad
learning_rate learning rate
momentum momentum for optimizer
save_checkpoint_epochs The frequency to save checkpoint per epoch
gpu_status show gpu status
save_predict_fn callback for save results of predict save_predict_fn(ids, embeddings, dir, task_id)
spacial args for tf
参数
train_verbose
0 = silent
1 = progress bar
2 = one line per epoch
tensorboard_steps update tensorboard every steps
profile_batch int or pair
estimator_hooks_fn estimator hooks function for early stop (tf.estimator.experimental.stop_if_no_decrease_hook)
callbacks custom keras callbacks
hooks custom estimator hooks
custom_metric_fn custom estimator metric function
eval_exporters instance of tf.estimator.Exporter
eval_hooks for tf.estimator.EvalSpec
eval_throttle_secs throttle_secs arg for tf.estimator.EvalSpec
keep_checkpoint_max args for tf.estimator.RunConfig
other args for input_fn
重载 galileo.framework.tf.python.tf_trainer.TFTrainer .
由 Doyxgen 通过分析 Galileo 的 源代码自动生成.