File tree Expand file tree Collapse file tree 2 files changed +8
-5
lines changed
api/src/main/java/com/theokanning/openai/fine_tuning
service/src/test/java/com/theokanning/openai/service Expand file tree Collapse file tree 2 files changed +8
-5
lines changed Original file line number Diff line number Diff line change 1717@ Data
1818public class Hyperparameters {
1919 /**
20+ * String or Integer type
2021 * The number of epochs to train the model for.
2122 * An epoch refers to one full cycle through the training dataset.
2223 * "Auto" decides the optimal number of epochs based on the size of the dataset.
2324 * If setting the number manually, we support any number between 1 and 50 epochs.
2425 */
2526 @ JsonProperty ("n_epochs" )
26- Integer nEpochs ;
27+ Object nEpochs ;
2728
2829 /**
30+ * String or Integer type
2931 * Number of examples in each batch. A larger batch size means that model parameters are updated less frequently, but with lower variance.
3032 */
3133 @ JsonProperty ("batch_size" )
32- Integer batchSize ;
34+ Object batchSize ;
3335
3436 /**
37+ * double or String type
3538 * Scaling factor for the learning rate. A smaller learning rate may be useful to avoid overfitting.
3639 */
3740 @ JsonProperty ("learning_rate_multiplier" )
38- Double learningRateMultiplier ;
41+ Object learningRateMultiplier ;
3942}
Original file line number Diff line number Diff line change @@ -44,7 +44,7 @@ void createFineTuningJob() {
4444 .build ();
4545 FineTuningJobRequest request = FineTuningJobRequest .builder ()
4646 .trainingFile (fileId )
47- .model ("gpt-4o-mini " )
47+ .model ("gpt-3.5-turbo " )
4848 .hyperparameters (hyperparameters )
4949 .build ();
5050
@@ -74,7 +74,7 @@ void listFineTuningEvents() {
7474 void retrieveFineTuningJob () {
7575 FineTuningJob fineTune = service .retrieveFineTuningJob (fineTuningJobId );
7676
77- assertTrue (fineTune .getModel ().startsWith ("gpt-4o-mini " ));
77+ assertTrue (fineTune .getModel ().startsWith ("gpt-3.5-turbo " ));
7878 }
7979
8080 @ Test
You can’t perform that action at this time.
0 commit comments