cavalab / brush

An interpretable machine learning library
http://cavalab.org/brush/
GNU General Public License v3.0
3 stars 0 forks source link

abide by pandas Dataframe types #56

Open lacava opened 6 months ago

lacava commented 6 months ago

Right now Brush Dataset tries to infer types, which might not be matched in another dataset for the same problem with, e.g., different numbers of integer values.

Solution: make Brush Dataset abide by pandas dataframe types as much as possible. Give the dataset initializer an optional argument of types for each feature.