objectstack/packages/spec/json-schema/ai/TrainingConfig.json at 8ae3d928f9c95dd320e89d8629a52085fc3d2d73 · objectstack-ai/objectstack · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
{
  "$ref": "#/definitions/TrainingConfig",
  "definitions": {
    "TrainingConfig": {
      "type": "object",
      "properties": {
        "trainingDataRatio": {
          "type": "number",
          "minimum": 0,
          "maximum": 1,
          "default": 0.8,
          "description": "Proportion of data for training"
        },
        "validationDataRatio": {
          "type": "number",
          "minimum": 0,
          "maximum": 1,
          "default": 0.1,
          "description": "Proportion for validation"
        },
        "testDataRatio": {
          "type": "number",
          "minimum": 0,
          "maximum": 1,
          "default": 0.1,
          "description": "Proportion for testing"
        },
        "dataFilter": {
          "type": "string",
          "description": "Formula to filter training data"
        },
        "minRecords": {
          "type": "integer",
          "default": 100,
          "description": "Minimum records required"
        },
        "maxRecords": {
          "type": "integer",
          "description": "Maximum records to use"
        },
        "strategy": {
          "type": "string",
          "enum": [
            "full",
            "incremental",
            "online",
            "transfer_learning"
          ],
          "default": "full"
        },
        "crossValidation": {
          "type": "boolean",
          "default": true
        },
        "folds": {
          "type": "integer",
          "minimum": 2,
          "maximum": 10,
          "default": 5,
          "description": "Cross-validation folds"
        },
        "earlyStoppingEnabled": {
          "type": "boolean",
          "default": true
        },
        "earlyStoppingPatience": {
          "type": "integer",
          "default": 10,
          "description": "Epochs without improvement before stopping"
        },
        "maxTrainingTime": {
          "type": "number",
          "description": "Maximum training time in seconds"
        },
        "gpuEnabled": {
          "type": "boolean",
          "default": false
        },
        "randomSeed": {
          "type": "integer",
          "description": "Random seed for reproducibility"
        }
      },
      "additionalProperties": false
    }
  },
  "$schema": "http://json-schema.org/draft-07/schema#"
}