Force features to be floats instead of doubles

google-research / exoplanet-ml

Machine learning models and utilities for exoplanet science.

Apache License 2.0

296 stars 117 forks source link

Force features to be floats instead of doubles #12

Open ritwik12 opened 4 years ago

ritwik12 commented 4 years ago

As per #7 , Make predictions give errors for various tensors as expected dtype double does not equal original dtype float. With suggestions from @caitlynlee Forcing tensors to be floats instead of doubles in astronet/predict.py solves the issue.


115: global_view =  preprocess.global_view(time, flux, FLAGS.period).astype(np.float32)
...
120: local_view = preprocess.local_view(time, flux, FLAGS.period, FLAGS.duration).astype(np.float32)

themison commented 4 years ago

dont work for me

ritwik12 commented 4 years ago

@themison What errors do you see?