Exemplo n.º 1
0
from part1_setup_data_normalization import load_data
from part2_load_model import loadmodel
from part1_train import train_cnn
from part1_statistics import plot_trainval_loss

data = load_data(batchsize=2)
network = loadmodel(15)

train = train_cnn(network.model, data)

print(train.train_loss_history)
print(train.val_loss_history)
plot_trainval_loss(train.train_loss_history, train.val_loss_history)
Exemplo n.º 2
0
from part1_setup_data_transformations import load_data
from part1_cnn import cnn
from part1_train import train_cnn
from part1_statistics import plot_trainval_loss
"""
load dataset
"""
import time
data = load_data()
network = cnn()

train = train_cnn(network, data)


print(train.train_loss_history)
print(train.val_loss_history)
plot_trainval_loss(train.train_loss_history, train.val_loss_history)

# print("--- %s seconds ---" % (time.time() - start_time))