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)[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)[source]

Module contents