search

gfd9111 / CCSNe_Detection

CCSNe detection project code from paper https://arxiv.org/abs/2410.06430
0 stars 0 forks source link

Example usage script #1

Open avivajpeyi opened 1 month ago

avivajpeyi commented 1 month ago

Hey! This looks great!

Do you have an example on how I can use this on some of my own timeseries data?

For example, 

model = load_CCSNE_Detection_model()
print(model.predict(timeseries))
gfd9111 commented 1 month ago

Hi Avi,

Thanks for your interest in running the prediction. I have uploaded a demo "CCSNE_demo.ipynb" that you can use. Please also make sure to download the model file "CCSNe_QNet_24Jan.onnx" from the Gdrive (link in the readme file).

Thanks, Steven