Python test_model_checkpointの例

コード例 #1

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

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

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

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

def BatchNormCNNTest(ckp_path, data_path):
    model_kwargs = {}
    auc, acc = test_model_checkpoint(ckp_path, BatchNormCNN, model_kwargs, 256,
                                     data_path)