ChEB-AI / python-chebai

GNU Affero General Public License v3.0
11 stars 4 forks source link

BCEWeighted loss fails on different pandas version #24

Closed MGlauer closed 3 months ago

MGlauer commented 3 months ago

Loading the pre-read data fails because the current implementation uses pickle.load to load a pandas dataframe.

Pandas is known to break its serialisation once in a while. Therefore, pandas.read_pickle should be used instead of pickle.load in order to load data frames.