how to predict with .edf file? about cardio

[dpodvyaznikov]

Hi, @Hunter0904 !

Predefined pipelines that you are using were designed to demonstrate how the library works on two specific tasks with specific datasets. If you take a look at dirichlet_predict_pipeline's source code, you'd see that fmt parameter in load action is set to 'wfdb', because the pipeline was developed to work with PhysioNet2017 Challenge data.

You can use our pipelines as an example to develop those suited for your needs. E.g., to solve your current issue you can just replace 'wfdb' to 'edf' in dirichlet_predict_pipeline.

[Hunter0904]

@dpodvyaznikov Thank for your answer.
But I don't have enough data to train new 'edf' model.
Is there a way to change the 'edf' file to 'wfdb' format ?
then I can use the same model to predict.

[dpodvyaznikov]

You still can use the model you've trained with dirichlet_train_pipeline.
The issue is in the way dirichlet_predict_pipeline tries to load the data. Since it has fmt='wfdb' in its load action, it tries to open file sample.hea instead of sample.edf. Since there is no such file, you get the error.
As I've said before, if you change fmt parameter to 'edf' in source code of dirichlet_predict_pipeline, you'll be able to run inference on edf files. Or you can make it a function argument and pass it directly when creatin pipeline.

[Hunter0904]

I will try it ~
very appreciate~thanks a lot

[dpodvyaznikov]

I'll close this issue for now.

[Hunter0904]

I had tried this. It's worked.
thanks a lot~~~