wurenzhi / learned_ndv_estimator Goto Github PK

Hi, I follow the readme page to generate training data but I found that I can not save f_s_list due to a data shape problem. I changed two parameters to obtain quick example results. My numpy version is 1.24.0 and the error that occurs in training_data_generation is shown as follows:

Besides, I found there is a bug in model_training that blocks the training loop,

. Line 55 declares three parameters for class ndvDatasetSparse but Line 80 does not provide the third parameter.

These two problems make generating training data and training the model not available. Could you please update the code when you are free, thank you very much.