Views
No views yet
1@inproceedings{gao2025machine,
2 title = {Machine Learning with Scarce Data: Ejection Fraction Prediction Using {PLAX} View},
3 author = {Gao, Zhiyuan and Yurk, Dominic and Abu-Mostafa, Yaser S.},
4 booktitle = {Medical Imaging with Deep Learning},
5 year = {2025},
6 url = {https://openreview.net/forum?id=JEN5FzeFZj}
7}1from huggingface_hub import hf_hub_download
2import torch, torchvision
3
4ckpt = hf_hub_download("Jeff4899/A4C_EF", "a4c_model.pth")
5model = torchvision.models.video.r2plus1d_18(weights=None)
6model.fc = torch.nn.Linear(model.fc.in_features, 1)
7state = torch.load(ckpt, map_location="cpu")
8model.load_state_dict(state, strict=False)
9model.eval()