Skip to content

Commit 4d413d2

Browse files
committed
fine tuning api接口类型兼容修改
1 parent 493a652 commit 4d413d2

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

api/src/main/java/com/theokanning/openai/fine_tuning/Hyperparameters.java

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,23 +17,26 @@
1717
@Data
1818
public 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
}

service/src/test/java/com/theokanning/openai/service/FineTuningTest.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)