Class FineTune

    • Constructor Detail

      • FineTune

        public FineTune()
    • Method Detail

      • setTrainingFile

        public void setTrainingFile​(String trainingFile)
      • setValidationFile

        public void setValidationFile​(String validationFile)
      • setModel

        public void setModel​(String model)
      • setN_epochs

        public void setN_epochs​(Integer n_epochs)
      • setBatchSize

        public void setBatchSize​(Integer batchSize)
      • setLearningRateMultiplier

        public void setLearningRateMultiplier​(Double learningRateMultiplier)
      • setPromptLossWeight

        public void setPromptLossWeight​(Double promptLossWeight)
      • setComputeClassificationMetrics

        public void setComputeClassificationMetrics​(boolean computeClassificationMetrics)
      • setClassificationNClasses

        public void setClassificationNClasses​(Integer classificationNClasses)
      • setClassificationBetas

        public void setClassificationBetas​(List classificationBetas)
      • setSuffix

        public void setSuffix​(String suffix)