ztlearn.dl.models package¶

Submodules¶

ztlearn.dl.models.sequential module¶

class ztlearn.dl.models.sequential.Sequential(init_method='he_normal', model_name='ztlearn_model')[source]¶

Bases: ztlearn.dl.models.trainer.Trainer

add(layer)[source]¶

added_layers¶

append_layer(layer)[source]¶

backward_pass(loss_grad, epoch_num, batch_num, batch_size)[source]¶

compile(loss='categorical_crossentropy', optimizer={})[source]¶

foward_pass(inputs, train_mode=False)[source]¶

init_method¶

is_trainable¶

layer_num¶

layers¶

model_name¶

summary(model_name='ztlearn_model')[source]¶

trainable¶

ztlearn.dl.models.trainer module¶

class ztlearn.dl.models.trainer.Trainer[source]¶

Bases: object

evaluate[source]¶

fit[source]¶

predict(sample_input, train_mode=False)[source]¶

test_on_batch(test_batch_data, test_batch_label, train_mode=False)[source]¶

train_on_batch(train_batch_data, train_batch_label, epoch_num=0, batch_num=0, batch_size=1)[source]¶

Module contents¶