예제 #1
0
def EfficientNetB4Test(ckp_path, data_path):
    model_kwargs = {"classes": 1}
    auc, acc = test_model_checkpoint(ckp_path, EfficientNetB4, model_kwargs,
                                     380, data_path)
    print(f"Model from {ckp_path}")
    print(f"auc = {auc}, acc = {acc}")
예제 #2
0
def BasicCNNTest(ckp_path, data_path):
    model_kwargs = {}
    auc, acc = test_model_checkpoint(ckp_path, BasicCNN, model_kwargs, 512,
                                     data_path)
    print(f"Model from {ckp_path}")
    print(f"auc = {auc}, acc = {acc}")
예제 #3
0
def VGG16Test(ckp_path, data_path):
    model_kwargs = {"in_channels": 3, "output_size": 1}
    auc, acc = test_model_checkpoint(ckp_path, VGG16, model_kwargs, 244,
                                     data_path)
    print(f"Model from {ckp_path}")
    print(f"auc = {auc}, acc = {acc}")
예제 #4
0
def ResNet152Test(ckp_path, data_path):
    model_kwargs = {"in_channels": 3, "classes": 1}
    auc, acc = test_model_checkpoint(ckp_path, ResNet152, model_kwargs, 244,
                                     data_path)
    print(f"Model from {ckp_path}")
    print(f"auc = {auc}, acc = {acc}")
예제 #5
0
def BatchNormCNNTest(ckp_path, data_path):
    model_kwargs = {}
    auc, acc = test_model_checkpoint(ckp_path, BatchNormCNN, model_kwargs, 256,
                                     data_path)