@elaraai/east-py-datascience 1.0.1 → 1.0.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (81) hide show
  1. package/dist/src/alns/alns.d.ts +528 -0
  2. package/dist/src/alns/alns.d.ts.map +1 -0
  3. package/dist/src/alns/alns.js +238 -0
  4. package/dist/src/alns/alns.js.map +1 -0
  5. package/dist/src/google_or/google_or.d.ts +2457 -0
  6. package/dist/src/google_or/google_or.d.ts.map +1 -0
  7. package/dist/src/google_or/google_or.js +547 -0
  8. package/dist/src/google_or/google_or.js.map +1 -0
  9. package/dist/src/gp/gp.d.ts +447 -0
  10. package/dist/src/gp/gp.d.ts.map +1 -0
  11. package/dist/src/gp/gp.js +222 -0
  12. package/dist/src/gp/gp.js.map +1 -0
  13. package/dist/src/index.d.ts +34 -0
  14. package/dist/src/index.d.ts.map +1 -0
  15. package/dist/src/index.js +57 -0
  16. package/dist/src/index.js.map +1 -0
  17. package/dist/src/lightgbm/lightgbm.d.ts +575 -0
  18. package/dist/src/lightgbm/lightgbm.d.ts.map +1 -0
  19. package/dist/src/lightgbm/lightgbm.js +241 -0
  20. package/dist/src/lightgbm/lightgbm.js.map +1 -0
  21. package/dist/src/lightning/lightning.d.ts +1594 -0
  22. package/dist/src/lightning/lightning.d.ts.map +1 -0
  23. package/dist/src/lightning/lightning.js +468 -0
  24. package/dist/src/lightning/lightning.js.map +1 -0
  25. package/dist/src/mads/mads.d.ts +410 -0
  26. package/dist/src/mads/mads.d.ts.map +1 -0
  27. package/dist/src/mads/mads.js +219 -0
  28. package/dist/src/mads/mads.js.map +1 -0
  29. package/dist/src/mapie/mapie.d.ts +3680 -0
  30. package/dist/src/mapie/mapie.d.ts.map +1 -0
  31. package/dist/src/mapie/mapie.js +616 -0
  32. package/dist/src/mapie/mapie.js.map +1 -0
  33. package/dist/src/ngboost/ngboost.d.ts +483 -0
  34. package/dist/src/ngboost/ngboost.d.ts.map +1 -0
  35. package/dist/src/ngboost/ngboost.js +231 -0
  36. package/dist/src/ngboost/ngboost.js.map +1 -0
  37. package/dist/src/optimization/optimization.d.ts +700 -0
  38. package/dist/src/optimization/optimization.d.ts.map +1 -0
  39. package/dist/src/optimization/optimization.js +411 -0
  40. package/dist/src/optimization/optimization.js.map +1 -0
  41. package/dist/src/optuna/optuna.d.ts +857 -0
  42. package/dist/src/optuna/optuna.d.ts.map +1 -0
  43. package/dist/src/optuna/optuna.js +270 -0
  44. package/dist/src/optuna/optuna.js.map +1 -0
  45. package/dist/src/pymc/pymc.d.ts +2932 -0
  46. package/dist/src/pymc/pymc.d.ts.map +1 -0
  47. package/dist/src/pymc/pymc.js +688 -0
  48. package/dist/src/pymc/pymc.js.map +1 -0
  49. package/dist/src/scipy/scipy.d.ts +2205 -0
  50. package/dist/src/scipy/scipy.d.ts.map +1 -0
  51. package/dist/src/scipy/scipy.js +884 -0
  52. package/dist/src/scipy/scipy.js.map +1 -0
  53. package/dist/src/shap/shap.d.ts +4611 -0
  54. package/dist/src/shap/shap.d.ts.map +1 -0
  55. package/dist/src/shap/shap.js +552 -0
  56. package/dist/src/shap/shap.js.map +1 -0
  57. package/dist/src/simanneal/simanneal.d.ts +628 -0
  58. package/dist/src/simanneal/simanneal.d.ts.map +1 -0
  59. package/dist/src/simanneal/simanneal.js +328 -0
  60. package/dist/src/simanneal/simanneal.js.map +1 -0
  61. package/dist/src/simulation/simulation.d.ts +224 -0
  62. package/dist/src/simulation/simulation.d.ts.map +1 -0
  63. package/dist/src/simulation/simulation.js +189 -0
  64. package/dist/src/simulation/simulation.js.map +1 -0
  65. package/dist/src/sklearn/sklearn.d.ts +6362 -0
  66. package/dist/src/sklearn/sklearn.d.ts.map +1 -0
  67. package/dist/src/sklearn/sklearn.js +1508 -0
  68. package/dist/src/sklearn/sklearn.js.map +1 -0
  69. package/dist/src/torch/torch.d.ts +1205 -0
  70. package/dist/src/torch/torch.d.ts.map +1 -0
  71. package/dist/src/torch/torch.js +440 -0
  72. package/dist/src/torch/torch.js.map +1 -0
  73. package/dist/src/types.d.ts +43 -0
  74. package/dist/src/types.d.ts.map +1 -0
  75. package/dist/src/types.js +44 -0
  76. package/dist/src/types.js.map +1 -0
  77. package/dist/src/xgboost/xgboost.d.ts +1424 -0
  78. package/dist/src/xgboost/xgboost.d.ts.map +1 -0
  79. package/dist/src/xgboost/xgboost.js +432 -0
  80. package/dist/src/xgboost/xgboost.js.map +1 -0
  81. package/package.json +3 -3
@@ -0,0 +1 @@
1
+ {"version":3,"file":"optuna.d.ts","sourceRoot":"","sources":["../../../src/optuna/optuna.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;;;;GAUG;AAEH,OAAO,EAEH,UAAU,EACV,WAAW,EACX,UAAU,EACV,SAAS,EACT,WAAW,EACX,SAAS,EACT,UAAU,EACV,WAAW,EACX,YAAY,EACZ,QAAQ,EACX,MAAM,eAAe,CAAC;AAMvB;;GAEG;AACH,eAAO,MAAM,cAAc;;;;;EAKzB,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,kBAAkB;IAC3B,6CAA6C;;IAE7C,2CAA2C;;IAE3C,yCAAyC;;IAEzC,6DAA6D;;EAE/D,CAAC;AAEH;;;;GAIG;AACH,eAAO,MAAM,cAAc;IACvB,qBAAqB;;IAErB,4DAA4D;;QAlB5D,6CAA6C;;QAE7C,2CAA2C;;QAE3C,yCAAyC;;QAEzC,6DAA6D;;;IAc7D,gDAAgD;;IAEhD,gDAAgD;;IAEhD,gCAAgC;;;;;;;EAElC,CAAC;AAEH;;;;GAIG;AACH,eAAO,MAAM,cAAc;IACvB,qBAAqB;;IAErB,sBAAsB;;;;;;;EAExB,CAAC;AAMH;;GAEG;AACH,eAAO,MAAM,yBAAyB;;;EAGpC,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,UAAU;IACnB,iBAAiB;;IAEjB,+DAA+D;;IAE/D,mDAAmD;;EAErD,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,qBAAqB;IAC9B,iDAAiD;;;;;IAEjD,8BAA8B;;IAE9B,sCAAsC;;IAEtC,gDAAgD;;QAlBhD,iBAAiB;;QAEjB,+DAA+D;;QAE/D,mDAAmD;;;IAgBnD,kEAAkE;;QA1ClE,qBAAqB;;QAErB,sBAAsB;;;;;;;;EA0CxB,CAAC;AAMH;;GAEG;AACH,eAAO,MAAM,eAAe;IACxB,eAAe;;IAEf,oCAAoC;;QAxDpC,qBAAqB;;QAErB,sBAAsB;;;;;;;;IAwDtB,sBAAsB;;EAExB,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,eAAe;IACxB,4BAA4B;;QAlE5B,qBAAqB;;QAErB,sBAAsB;;;;;;;;IAkEtB,2BAA2B;;IAE3B,2BAA2B;;QAhB3B,eAAe;;QAEf,oCAAoC;;YAxDpC,qBAAqB;;YAErB,sBAAsB;;;;;;;;QAwDtB,sBAAsB;;;EAcxB,CAAC;AAMH;;;;;GAKG;AACH,eAAO,MAAM,qBAAqB;IApF9B,qBAAqB;;IAErB,sBAAsB;;;;;;;gBAqFzB,CAAC;AAMF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAmDG;AACH,eAAO,MAAM,eAAe;IAnKxB,qBAAqB;;IAErB,4DAA4D;;QAlB5D,6CAA6C;;QAE7C,2CAA2C;;QAE3C,yCAAyC;;QAEzC,6DAA6D;;;IAc7D,gDAAgD;;IAEhD,gDAAgD;;IAEhD,gCAAgC;;;;;;;;IAUhC,qBAAqB;;IAErB,sBAAsB;;;;;;;;IAgCtB,iDAAiD;;;;;IAEjD,8BAA8B;;IAE9B,sCAAsC;;IAEtC,gDAAgD;;QAlBhD,iBAAiB;;QAEjB,+DAA+D;;QAE/D,mDAAmD;;;IAgBnD,kEAAkE;;QA1ClE,qBAAqB;;QAErB,sBAAsB;;;;;;;;;IAgEtB,4BAA4B;;QAlE5B,qBAAqB;;QAErB,sBAAsB;;;;;;;;IAkEtB,2BAA2B;;IAE3B,2BAA2B;;QAhB3B,eAAe;;QAEf,oCAAoC;;YAxDpC,qBAAqB;;YAErB,sBAAsB;;;;;;;;QAwDtB,sBAAsB;;;GA+FzB,CAAC;AAMF;;GAEG;AACH,eAAO,MAAM,WAAW;IACpB,mCAAmC;;;;;;;IAEnC,gCAAgC;;QAvMhC,6CAA6C;;QAE7C,2CAA2C;;QAE3C,yCAAyC;;QAEzC,6DAA6D;;;IAmM7D,sCAAsC;;QAzLtC,qBAAqB;;QAErB,4DAA4D;;YAlB5D,6CAA6C;;YAE7C,2CAA2C;;YAE3C,yCAAyC;;YAEzC,6DAA6D;;;QAc7D,gDAAgD;;QAEhD,gDAAgD;;QAEhD,gCAAgC;;;;;;;;IAmLhC,2BAA2B;;QAzK3B,qBAAqB;;QAErB,sBAAsB;;;;;;;;IAyKtB,kCAAkC;;;;;IAElC,kBAAkB;;QAvJlB,iBAAiB;;QAEjB,+DAA+D;;QAE/D,mDAAmD;;;IAqJnD,wBAAwB;;QA7IxB,iDAAiD;;;;;QAEjD,8BAA8B;;QAE9B,sCAAsC;;QAEtC,gDAAgD;;YAlBhD,iBAAiB;;YAEjB,+DAA+D;;YAE/D,mDAAmD;;;QAgBnD,kEAAkE;;YA1ClE,qBAAqB;;YAErB,sBAAsB;;;;;;;;;IA+KtB,wBAAwB;;QA3HxB,eAAe;;QAEf,oCAAoC;;YAxDpC,qBAAqB;;YAErB,sBAAsB;;;;;;;;QAwDtB,sBAAsB;;;IAyHtB,wBAAwB;;QAjHxB,4BAA4B;;YAlE5B,qBAAqB;;YAErB,sBAAsB;;;;;;;;QAkEtB,2BAA2B;;QAE3B,2BAA2B;;YAhB3B,eAAe;;YAEf,oCAAoC;;gBAxDpC,qBAAqB;;gBAErB,sBAAsB;;;;;;;;YAwDtB,sBAAsB;;;;IA2HtB,8BAA8B;;QArL9B,qBAAqB;;QAErB,sBAAsB;;;;;;;;CAqLhB,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,eAAO,MAAM,MAAM;IACf;;;;OAIG;;QA7OH,qBAAqB;;QAErB,4DAA4D;;YAlB5D,6CAA6C;;YAE7C,2CAA2C;;YAE3C,yCAAyC;;YAEzC,6DAA6D;;;QAc7D,gDAAgD;;QAEhD,gDAAgD;;QAEhD,gCAAgC;;;;;;;;QAUhC,qBAAqB;;QAErB,sBAAsB;;;;;;;;QAgCtB,iDAAiD;;;;;QAEjD,8BAA8B;;QAE9B,sCAAsC;;QAEtC,gDAAgD;;YAlBhD,iBAAiB;;YAEjB,+DAA+D;;YAE/D,mDAAmD;;;QAgBnD,kEAAkE;;YA1ClE,qBAAqB;;YAErB,sBAAsB;;;;;;;;;QAgEtB,4BAA4B;;YAlE5B,qBAAqB;;YAErB,sBAAsB;;;;;;;;QAkEtB,2BAA2B;;QAE3B,2BAA2B;;YAhB3B,eAAe;;YAEf,oCAAoC;;gBAxDpC,qBAAqB;;gBAErB,sBAAsB;;;;;;;;YAwDtB,sBAAsB;;;;IAoKtB;;OAEG;;QA7DH,mCAAmC;;;;;;;QAEnC,gCAAgC;;YAvMhC,6CAA6C;;YAE7C,2CAA2C;;YAE3C,yCAAyC;;YAEzC,6DAA6D;;;QAmM7D,sCAAsC;;YAzLtC,qBAAqB;;YAErB,4DAA4D;;gBAlB5D,6CAA6C;;gBAE7C,2CAA2C;;gBAE3C,yCAAyC;;gBAEzC,6DAA6D;;;YAc7D,gDAAgD;;YAEhD,gDAAgD;;YAEhD,gCAAgC;;;;;;;;QAmLhC,2BAA2B;;YAzK3B,qBAAqB;;YAErB,sBAAsB;;;;;;;;QAyKtB,kCAAkC;;;;;QAElC,kBAAkB;;YAvJlB,iBAAiB;;YAEjB,+DAA+D;;YAE/D,mDAAmD;;;QAqJnD,wBAAwB;;YA7IxB,iDAAiD;;;;;YAEjD,8BAA8B;;YAE9B,sCAAsC;;YAEtC,gDAAgD;;gBAlBhD,iBAAiB;;gBAEjB,+DAA+D;;gBAE/D,mDAAmD;;;YAgBnD,kEAAkE;;gBA1ClE,qBAAqB;;gBAErB,sBAAsB;;;;;;;;;QA+KtB,wBAAwB;;YA3HxB,eAAe;;YAEf,oCAAoC;;gBAxDpC,qBAAqB;;gBAErB,sBAAsB;;;;;;;;YAwDtB,sBAAsB;;;QAyHtB,wBAAwB;;YAjHxB,4BAA4B;;gBAlE5B,qBAAqB;;gBAErB,sBAAsB;;;;;;;;YAkEtB,2BAA2B;;YAE3B,2BAA2B;;gBAhB3B,eAAe;;gBAEf,oCAAoC;;oBAxDpC,qBAAqB;;oBAErB,sBAAsB;;;;;;;;gBAwDtB,sBAAsB;;;;QA2HtB,8BAA8B;;YArL9B,qBAAqB;;YAErB,sBAAsB;;;;;;;;;CAgOhB,CAAC"}
@@ -0,0 +1,270 @@
1
+ /**
2
+ * Copyright (c) 2025 Elara AI Pty Ltd
3
+ * Dual-licensed under AGPL-3.0 and commercial license. See LICENSE for details.
4
+ */
5
+ /**
6
+ * Bayesian optimization using Optuna's TPE sampler.
7
+ *
8
+ * Provides parameter optimization for blackbox functions with mixed parameter
9
+ * types (categorical, integer, continuous). Useful for:
10
+ * - Hyperparameter tuning for ML models
11
+ * - Design optimization
12
+ * - Any blackbox optimization with structured parameter spaces
13
+ *
14
+ * @packageDocumentation
15
+ */
16
+ import { East, StructType, VariantType, OptionType, ArrayType, IntegerType, FloatType, StringType, BooleanType, FunctionType, NullType, } from "@elaraai/east";
17
+ // ===========================================
18
+ // Parameter Types
19
+ // ===========================================
20
+ /**
21
+ * Parameter value type (can be int, float, string, or bool).
22
+ */
23
+ export const ParamValueType = VariantType({
24
+ int: IntegerType,
25
+ float: FloatType,
26
+ string: StringType,
27
+ bool: BooleanType,
28
+ });
29
+ /**
30
+ * Parameter space kind for defining search spaces.
31
+ */
32
+ export const ParamSpaceKindType = VariantType({
33
+ /** Integer parameter with low/high bounds */
34
+ int: NullType,
35
+ /** Float parameter with low/high bounds */
36
+ float: NullType,
37
+ /** Categorical parameter with choices */
38
+ categorical: NullType,
39
+ /** Log-uniform float parameter (for learning rates, etc.) */
40
+ log_uniform: NullType,
41
+ });
42
+ /**
43
+ * Parameter search space definition.
44
+ *
45
+ * Defines a single parameter's name, type, and valid range/choices.
46
+ */
47
+ export const ParamSpaceType = StructType({
48
+ /** Parameter name */
49
+ name: StringType,
50
+ /** Parameter kind (int, float, categorical, log_uniform) */
51
+ kind: ParamSpaceKindType,
52
+ /** Lower bound (for int, float, log_uniform) */
53
+ low: OptionType(FloatType),
54
+ /** Upper bound (for int, float, log_uniform) */
55
+ high: OptionType(FloatType),
56
+ /** Choices (for categorical) */
57
+ choices: OptionType(ArrayType(ParamValueType)),
58
+ });
59
+ /**
60
+ * Named parameter (name + value pair).
61
+ *
62
+ * Represents a single parameter with its suggested/best value.
63
+ */
64
+ export const NamedParamType = StructType({
65
+ /** Parameter name */
66
+ name: StringType,
67
+ /** Parameter value */
68
+ value: ParamValueType,
69
+ });
70
+ // ===========================================
71
+ // Optimization Config Types
72
+ // ===========================================
73
+ /**
74
+ * Optimization direction (minimize or maximize).
75
+ */
76
+ export const OptimizationDirectionType = VariantType({
77
+ minimize: NullType,
78
+ maximize: NullType,
79
+ });
80
+ /**
81
+ * Pruner type for early stopping of unpromising trials.
82
+ */
83
+ export const PrunerType = VariantType({
84
+ /** No pruning */
85
+ none: NullType,
86
+ /** Median pruner - prune if below median of previous trials */
87
+ median: NullType,
88
+ /** Hyperband pruner - aggressive early stopping */
89
+ hyperband: NullType,
90
+ });
91
+ /**
92
+ * Optuna study configuration.
93
+ */
94
+ export const OptunaStudyConfigType = StructType({
95
+ /** Optimization direction (default: minimize) */
96
+ direction: OptionType(OptimizationDirectionType),
97
+ /** Number of trials to run */
98
+ n_trials: IntegerType,
99
+ /** Random seed for reproducibility */
100
+ random_state: OptionType(IntegerType),
101
+ /** Pruner for early stopping (default: none) */
102
+ pruner: OptionType(PrunerType),
103
+ /** Initial parameters to evaluate before sampling (warm-start) */
104
+ initial_params: OptionType(ArrayType(NamedParamType)),
105
+ });
106
+ // ===========================================
107
+ // Result Types
108
+ // ===========================================
109
+ /**
110
+ * Single trial result.
111
+ */
112
+ export const TrialResultType = StructType({
113
+ /** Trial ID */
114
+ trial_id: IntegerType,
115
+ /** Parameters used in this trial */
116
+ params: ArrayType(NamedParamType),
117
+ /** Objective score */
118
+ score: FloatType,
119
+ });
120
+ /**
121
+ * Optimization study result.
122
+ */
123
+ export const StudyResultType = StructType({
124
+ /** Best parameters found */
125
+ best_params: ArrayType(NamedParamType),
126
+ /** Best objective score */
127
+ best_score: FloatType,
128
+ /** All completed trials */
129
+ trials: ArrayType(TrialResultType),
130
+ });
131
+ // ===========================================
132
+ // Objective Function Type
133
+ // ===========================================
134
+ /**
135
+ * Objective function type: takes named params, returns score.
136
+ *
137
+ * The objective function is an East function that receives suggested
138
+ * parameters and returns a scalar score to minimize/maximize.
139
+ */
140
+ export const ObjectiveFunctionType = FunctionType([ArrayType(NamedParamType)], FloatType);
141
+ // ===========================================
142
+ // Platform Functions
143
+ // ===========================================
144
+ /**
145
+ * Run Bayesian optimization with Optuna.
146
+ *
147
+ * Uses TPE (Tree-structured Parzen Estimator) to efficiently search
148
+ * the parameter space, balancing exploration and exploitation.
149
+ *
150
+ * @param search_space - Array of parameter space definitions
151
+ * @param objective - East function that takes params and returns score
152
+ * @param config - Study configuration (n_trials, direction, etc.)
153
+ * @returns Study result with best params, best score, and all trials
154
+ *
155
+ * @example
156
+ * ```ts
157
+ * import { East, FloatType, variant } from "@elaraai/east";
158
+ * import { Optuna } from "@elaraai/east-py-datascience";
159
+ *
160
+ * // Define objective: minimize (x - 2)^2 + (y - 3)^2
161
+ * const objective = East.function(
162
+ * [ArrayType(Optuna.Types.NamedParamType)],
163
+ * FloatType,
164
+ * ($, params) => {
165
+ * // Extract params by name
166
+ * const x = $.let(params.get(0n).value); // First param
167
+ * const y = $.let(params.get(1n).value); // Second param
168
+ * // Compute objective (assuming float params)
169
+ * return $.return(
170
+ * x.match({ float: v => v }).subtract(2.0).multiply(
171
+ * x.match({ float: v => v }).subtract(2.0)
172
+ * ).add(
173
+ * y.match({ float: v => v }).subtract(3.0).multiply(
174
+ * y.match({ float: v => v }).subtract(3.0)
175
+ * )
176
+ * )
177
+ * );
178
+ * }
179
+ * );
180
+ *
181
+ * const optimize = East.function([], Optuna.Types.StudyResultType, $ => {
182
+ * const search_space = $.let([
183
+ * { name: "x", kind: variant("float", null), low: variant("some", 0.0), high: variant("some", 5.0), choices: variant("none", null) },
184
+ * { name: "y", kind: variant("float", null), low: variant("some", 0.0), high: variant("some", 5.0), choices: variant("none", null) },
185
+ * ]);
186
+ * const config = $.let({
187
+ * direction: variant("some", variant("minimize", null)),
188
+ * n_trials: 50n,
189
+ * random_state: variant("some", 42n),
190
+ * pruner: variant("none", null),
191
+ * });
192
+ * return $.return(Optuna.optimize(search_space, objective, config));
193
+ * });
194
+ * ```
195
+ */
196
+ export const optuna_optimize = East.platform("optuna_optimize", [
197
+ ArrayType(ParamSpaceType),
198
+ ObjectiveFunctionType,
199
+ OptunaStudyConfigType,
200
+ ], StudyResultType);
201
+ // ===========================================
202
+ // Grouped Export
203
+ // ===========================================
204
+ /**
205
+ * Type definitions for Optuna optimization.
206
+ */
207
+ export const OptunaTypes = {
208
+ /** Parameter value variant type */
209
+ ParamValueType,
210
+ /** Parameter space kind type */
211
+ ParamSpaceKindType,
212
+ /** Parameter space definition type */
213
+ ParamSpaceType,
214
+ /** Named parameter type */
215
+ NamedParamType,
216
+ /** Optimization direction type */
217
+ OptimizationDirectionType,
218
+ /** Pruner type */
219
+ PrunerType,
220
+ /** Study config type */
221
+ StudyConfigType: OptunaStudyConfigType,
222
+ /** Trial result type */
223
+ TrialResultType,
224
+ /** Study result type */
225
+ StudyResultType,
226
+ /** Objective function type */
227
+ ObjectiveFunctionType,
228
+ };
229
+ /**
230
+ * Bayesian optimization using Optuna.
231
+ *
232
+ * Provides efficient parameter optimization using TPE (Tree-structured
233
+ * Parzen Estimator) for blackbox functions with mixed parameter types.
234
+ *
235
+ * Supports:
236
+ * - Integer, float, and categorical parameters
237
+ * - Log-uniform sampling for learning rates
238
+ * - Minimization and maximization
239
+ * - Early stopping with pruners
240
+ *
241
+ * @example
242
+ * ```ts
243
+ * import { East, FloatType, variant } from "@elaraai/east";
244
+ * import { Optuna } from "@elaraai/east-py-datascience";
245
+ *
246
+ * const objective = East.function(
247
+ * [ArrayType(Optuna.Types.NamedParamType)],
248
+ * FloatType,
249
+ * ($, params) => {
250
+ * // Your objective function here
251
+ * return $.return(score);
252
+ * }
253
+ * );
254
+ *
255
+ * const result = Optuna.optimize(search_space, objective, config);
256
+ * ```
257
+ */
258
+ export const Optuna = {
259
+ /**
260
+ * Run Bayesian optimization.
261
+ *
262
+ * Efficiently searches the parameter space using TPE.
263
+ */
264
+ optimize: optuna_optimize,
265
+ /**
266
+ * Type definitions for Optuna functions.
267
+ */
268
+ Types: OptunaTypes,
269
+ };
270
+ //# sourceMappingURL=optuna.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"optuna.js","sourceRoot":"","sources":["../../../src/optuna/optuna.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;;;;GAUG;AAEH,OAAO,EACH,IAAI,EACJ,UAAU,EACV,WAAW,EACX,UAAU,EACV,SAAS,EACT,WAAW,EACX,SAAS,EACT,UAAU,EACV,WAAW,EACX,YAAY,EACZ,QAAQ,GACX,MAAM,eAAe,CAAC;AAEvB,8CAA8C;AAC9C,kBAAkB;AAClB,8CAA8C;AAE9C;;GAEG;AACH,MAAM,CAAC,MAAM,cAAc,GAAG,WAAW,CAAC;IACtC,GAAG,EAAE,WAAW;IAChB,KAAK,EAAE,SAAS;IAChB,MAAM,EAAE,UAAU;IAClB,IAAI,EAAE,WAAW;CACpB,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,WAAW,CAAC;IAC1C,6CAA6C;IAC7C,GAAG,EAAE,QAAQ;IACb,2CAA2C;IAC3C,KAAK,EAAE,QAAQ;IACf,yCAAyC;IACzC,WAAW,EAAE,QAAQ;IACrB,6DAA6D;IAC7D,WAAW,EAAE,QAAQ;CACxB,CAAC,CAAC;AAEH;;;;GAIG;AACH,MAAM,CAAC,MAAM,cAAc,GAAG,UAAU,CAAC;IACrC,qBAAqB;IACrB,IAAI,EAAE,UAAU;IAChB,4DAA4D;IAC5D,IAAI,EAAE,kBAAkB;IACxB,gDAAgD;IAChD,GAAG,EAAE,UAAU,CAAC,SAAS,CAAC;IAC1B,gDAAgD;IAChD,IAAI,EAAE,UAAU,CAAC,SAAS,CAAC;IAC3B,gCAAgC;IAChC,OAAO,EAAE,UAAU,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;CACjD,CAAC,CAAC;AAEH;;;;GAIG;AACH,MAAM,CAAC,MAAM,cAAc,GAAG,UAAU,CAAC;IACrC,qBAAqB;IACrB,IAAI,EAAE,UAAU;IAChB,sBAAsB;IACtB,KAAK,EAAE,cAAc;CACxB,CAAC,CAAC;AAEH,8CAA8C;AAC9C,4BAA4B;AAC5B,8CAA8C;AAE9C;;GAEG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,WAAW,CAAC;IACjD,QAAQ,EAAE,QAAQ;IAClB,QAAQ,EAAE,QAAQ;CACrB,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG,WAAW,CAAC;IAClC,iBAAiB;IACjB,IAAI,EAAE,QAAQ;IACd,+DAA+D;IAC/D,MAAM,EAAE,QAAQ;IAChB,mDAAmD;IACnD,SAAS,EAAE,QAAQ;CACtB,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,UAAU,CAAC;IAC5C,iDAAiD;IACjD,SAAS,EAAE,UAAU,CAAC,yBAAyB,CAAC;IAChD,8BAA8B;IAC9B,QAAQ,EAAE,WAAW;IACrB,sCAAsC;IACtC,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;IACrC,gDAAgD;IAChD,MAAM,EAAE,UAAU,CAAC,UAAU,CAAC;IAC9B,kEAAkE;IAClE,cAAc,EAAE,UAAU,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;CACxD,CAAC,CAAC;AAEH,8CAA8C;AAC9C,eAAe;AACf,8CAA8C;AAE9C;;GAEG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,UAAU,CAAC;IACtC,eAAe;IACf,QAAQ,EAAE,WAAW;IACrB,oCAAoC;IACpC,MAAM,EAAE,SAAS,CAAC,cAAc,CAAC;IACjC,sBAAsB;IACtB,KAAK,EAAE,SAAS;CACnB,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,UAAU,CAAC;IACtC,4BAA4B;IAC5B,WAAW,EAAE,SAAS,CAAC,cAAc,CAAC;IACtC,2BAA2B;IAC3B,UAAU,EAAE,SAAS;IACrB,2BAA2B;IAC3B,MAAM,EAAE,SAAS,CAAC,eAAe,CAAC;CACrC,CAAC,CAAC;AAEH,8CAA8C;AAC9C,0BAA0B;AAC1B,8CAA8C;AAE9C;;;;;GAKG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,YAAY,CAC7C,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,EAC3B,SAAS,CACZ,CAAC;AAEF,8CAA8C;AAC9C,qBAAqB;AACrB,8CAA8C;AAE9C;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAmDG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,IAAI,CAAC,QAAQ,CACxC,iBAAiB,EACjB;IACI,SAAS,CAAC,cAAc,CAAC;IACzB,qBAAqB;IACrB,qBAAqB;CACxB,EACD,eAAe,CAClB,CAAC;AAEF,8CAA8C;AAC9C,iBAAiB;AACjB,8CAA8C;AAE9C;;GAEG;AACH,MAAM,CAAC,MAAM,WAAW,GAAG;IACvB,mCAAmC;IACnC,cAAc;IACd,gCAAgC;IAChC,kBAAkB;IAClB,sCAAsC;IACtC,cAAc;IACd,2BAA2B;IAC3B,cAAc;IACd,kCAAkC;IAClC,yBAAyB;IACzB,kBAAkB;IAClB,UAAU;IACV,wBAAwB;IACxB,eAAe,EAAE,qBAAqB;IACtC,wBAAwB;IACxB,eAAe;IACf,wBAAwB;IACxB,eAAe;IACf,8BAA8B;IAC9B,qBAAqB;CACf,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,MAAM,CAAC,MAAM,MAAM,GAAG;IAClB;;;;OAIG;IACH,QAAQ,EAAE,eAAe;IAEzB;;OAEG;IACH,KAAK,EAAE,WAAW;CACZ,CAAC"}