| __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.') | |