Exemplo n.º 1
0
def run_test():
    # model = MyModel.load_from_metrics(
    #     weights_path='experiments/saved_batch_attn_rnn/model_weights/mimic_all/1/_ckpt_epoch_10.ckpt',
    #     tags_csv='experiments/saved_batch_attn_rnn/test_tube_data/mimic_all/version_1/meta_tags.csv',
    #     on_gpu=True,
    #     map_location=None
    # )
    #    model = MyModel.load_from_metrics(
    #        weights_path='saved/rel_interval/model_weights/inner/1/_ckpt_epoch_9.ckpt',
    #        tags_csv='saved/rel_interval/test_tube_data/inner/version_1/meta_tags.csv',
    #        on_gpu=True,
    #        map_location=torch.device('cuda')
    #    )
    #    model = MyModel.load_from_metrics(
    #        weights_path='saved/rel_interval/model_weights/nash/1/_ckpt_epoch_4.ckpt',
    #        tags_csv='saved/rel_interval/test_tube_data/nash/version_1/meta_tags.csv',
    #        on_gpu=True,
    #        map_location=torch.device('cuda')
    #    )
    #    model = MyModel.load_from_metrics(
    #        weights_path='saved/rel_interval/model_weights/nash/6/_ckpt_epoch_4.ckpt',
    #        tags_csv='saved/rel_interval/test_tube_data/nash/version_6/meta_tags.csv',
    #        on_gpu=True,
    #        map_location=torch.device('cuda')
    #    )
    #    trainer = Trainer()
    #    trainer.test(model)
    #    model = MyModel.load_from_metrics(
    #        weights_path='saved/rel_interval/model_weights/nash/7/_ckpt_epoch_2.ckpt',
    #        tags_csv='saved/rel_interval/test_tube_data/nash/version_7/meta_tags.csv',
    #        on_gpu=True,
    #        map_location=torch.device('cuda')
    #    )
    #    trainer = Trainer()
    #    trainer.test(model)
    model = MyModel.load_from_metrics(
        weights_path=
        'saved_seed1/rel_interval_layer_varient/model_weights/ad/0/_ckpt_epoch_8.ckpt',
        tags_csv=
        'saved_seed1/rel_interval_layer_varient/test_tube_data/ad/version_0/meta_tags.csv',
        on_gpu=True,
        map_location=torch.device('cuda'))
    trainer = Trainer()
    trainer.test(model)
Exemplo n.º 2
0
def run_test():
    #    model = MyModel.load_from_metrics(
    #        weights_path='saved/retain/model_weights/inner/7/_ckpt_epoch_9.ckpt',
    #        tags_csv='saved/retain/test_tube_data/inner/version_7/meta_tags.csv',
    #        on_gpu=True,
    #        map_location=None,
    #    )
    #    model = MyModel.load_from_metrics(
    #        weights_path='saved/retain/model_weights/nash/2/_ckpt_epoch_22.ckpt',
    #        tags_csv='saved/retain/test_tube_data/nash/version_2/meta_tags.csv',
    #        on_gpu=True,
    #        map_location=None,
    #    )
    #    model = MyModel.load_from_metrics(
    #        weights_path='saved/retain/model_weights/nash/6/_ckpt_epoch_4.ckpt',
    #        tags_csv='saved/retain/test_tube_data/nash/version_6/meta_tags.csv',
    #        on_gpu=True,
    #        map_location=torch.device('cuda'),
    #    )
    #    trainer = Trainer()
    #    trainer.test(model)
    #    model = MyModel.load_from_metrics(
    #        weights_path='saved/retain/model_weights/nash/7/_ckpt_epoch_4.ckpt',
    #        tags_csv='saved/retain/test_tube_data/nash/version_7/meta_tags.csv',
    #        on_gpu=True,
    #        map_location=torch.device('cuda'),
    #    )
    #    trainer = Trainer()
    #    trainer.test(model)
    model = MyModel.load_from_metrics(
        weights_path='saved_seed1/retain/model_weights/ad/1/_ckpt_epoch_8.ckpt',
        tags_csv='saved_seed1/retain/test_tube_data/ad/version_1/meta_tags.csv',
        on_gpu=True,
        map_location=torch.device('cuda'),
    )
    trainer = Trainer()
    trainer.test(model)
def run_test():
    #    model = MyModel.load_from_metrics(
    #        weights_path='saved/tlstm/model_weights/nash/4/_ckpt_epoch_22.ckpt',
    #        tags_csv='saved/tlstm/test_tube_data/nash/version_4/meta_tags.csv',
    #        on_gpu=True,
    #        map_location=torch.device('cuda')
    #    )
    #    model = MyModel.load_from_metrics(
    #        weights_path='saved/tlstm/model_weights/ad/0/_ckpt_epoch_8.ckpt',
    #        tags_csv='saved/tlstm/test_tube_data/ad/version_0/meta_tags.csv',
    #        on_gpu=True,
    #        map_location=torch.device('cuda')
    #    )
    #    model = MyModel.load_from_metrics(
    #        weights_path='saved/tlstm/model_weights/nash/8/_ckpt_epoch_6.ckpt',
    #        tags_csv='saved/tlstm/test_tube_data/nash/version_10/meta_tags.csv',
    #        on_gpu=True,
    #        map_location=torch.device('cuda')
    #    )
    #    trainer = Trainer()
    #    trainer.test(model)
    #    model = MyModel.load_from_metrics(
    #        weights_path='saved/tlstm/model_weights/nash/9/_ckpt_epoch_6.ckpt',
    #        tags_csv='saved/tlstm/test_tube_data/nash/version_9/meta_tags.csv',
    #        on_gpu=True,
    #        map_location=torch.device('cuda')
    #    )
    #    trainer = Trainer()
    #    trainer.test(model)
    model = MyModel.load_from_metrics(
        weights_path='saved_seed1/tlstm/model_weights/ad/2/_ckpt_epoch_7.ckpt',
        tags_csv='saved_seed1/tlstm/test_tube_data/ad/version_2/meta_tags.csv',
        on_gpu=True,
        map_location=torch.device('cuda'))
    trainer = Trainer()
    trainer.test(model)