Map<String, dynamic> toMap() { return <String, dynamic>{ 'categoricalFeatures': categoricalFeatures, 'numericalFeatures': numericalFeatures, }; }