File size: 968 Bytes
6fc43ab |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
__version__ = '0.0.1'
from . import nn
from . import model
# # load pretrained transformer
# pretrained_transformer = model.Transformer.from_ckpt('{}/ckpt/ckpt.pt'.format(__path__[0]))
# from . import shap_adrd
# from .model import DynamicCalibratedClassifier
# from .model import StaticCalibratedClassifier
# load fitted transformer and calibrated wrapper
# try:
# fitted_resnet3d = model.CNNResNet3DWithLinearClassifier.from_ckpt('{}/ckpt/ckpt_img_072523.pt'.format(__path__[0]))
# fitted_calibrated_classifier_nonimg = StaticCalibratedClassifier.from_ckpt(
# filepath_state_dict = '{}/ckpt/static_calibrated_classifier_073023.pkl'.format(__path__[0]),
# filepath_wrapped_model = '{}/ckpt/ckpt_080823.pt'.format(__path__[0]),
# )
# fitted_transformer_nonimg = fitted_calibrated_classifier_nonimg.model
# shap_explainer = shap_adrd.SamplingExplainer(fitted_transformer_nonimg)
# except:
# print('Fail to load checkpoints.')
|