toMap method

Map<String, dynamic> toMap()

Implementation

Map<String, dynamic> toMap() {
  return <String, dynamic>{
    'model': model,
    'n_epochs': nEpochs,
    'batch_size': batchSize,
    'learning_rate_multiplier': learningRateMultiplier,
    'prompt_loss_weight': promptLossWeight,
    'compute_classification_metrics': computeClassificationMetrics,
    'classification_n_classes': classificationNClasses,
    'classificationPositiveClass': classificationPositiveClass,
    'suffix': suffix,
  };
}