Map<String, dynamic> toMap() { return <String, dynamic>{ 'labels': labels, 'lossType': lossType, 'modelType': modelType, }; }