weave-typescript 0.1.0

package/dist/weave/datamanagement/generate/v1/configuration.pb.js

@@ -0,0 +1,799 @@
+"use strict";
+// Code generated by protoc-gen-ts_proto. DO NOT EDIT.
+// versions:
+//   protoc-gen-ts_proto  v2.6.1
+//   protoc               unknown
+// source: weave/datamanagement/generate/v1/configuration.proto
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ModelConfig_CustomParametersEntry = exports.ModelConfig = exports.PrivacyConfig = exports.ColumnConstraint = exports.DataQualityConfig = exports.Configuration = exports.ModelType = exports.protobufPackage = void 0;
+exports.modelTypeFromJSON = modelTypeFromJSON;
+exports.modelTypeToJSON = modelTypeToJSON;
+/* eslint-disable */
+const wire_1 = require("@bufbuild/protobuf/wire");
+exports.protobufPackage = "weave.datamanagement.generate.v1";
+/** ModelType defines the available synthetic data generation models */
+var ModelType;
+(function (ModelType) {
+    ModelType[ModelType["MODEL_TYPE_UNSPECIFIED"] = 0] = "MODEL_TYPE_UNSPECIFIED";
+    /** MODEL_TYPE_CTGAN - Conditional Tabular GAN */
+    ModelType[ModelType["MODEL_TYPE_CTGAN"] = 1] = "MODEL_TYPE_CTGAN";
+    /** MODEL_TYPE_TVAE - Tabular Variational Autoencoder */
+    ModelType[ModelType["MODEL_TYPE_TVAE"] = 2] = "MODEL_TYPE_TVAE";
+    /** MODEL_TYPE_COPULA - Gaussian Copula */
+    ModelType[ModelType["MODEL_TYPE_COPULA"] = 3] = "MODEL_TYPE_COPULA";
+    /** MODEL_TYPE_FAST - Fast mode (less accurate but quicker) */
+    ModelType[ModelType["MODEL_TYPE_FAST"] = 4] = "MODEL_TYPE_FAST";
+    /** MODEL_TYPE_GAN - Generic GAN implementation */
+    ModelType[ModelType["MODEL_TYPE_GAN"] = 5] = "MODEL_TYPE_GAN";
+    /** MODEL_TYPE_WGAN - Wasserstein GAN */
+    ModelType[ModelType["MODEL_TYPE_WGAN"] = 6] = "MODEL_TYPE_WGAN";
+    /** MODEL_TYPE_SMOTE - Synthetic Minority Over-sampling Technique */
+    ModelType[ModelType["MODEL_TYPE_SMOTE"] = 7] = "MODEL_TYPE_SMOTE";
+    /** MODEL_TYPE_BAYESIAN - Bayesian Network */
+    ModelType[ModelType["MODEL_TYPE_BAYESIAN"] = 8] = "MODEL_TYPE_BAYESIAN";
+    /** MODEL_TYPE_HMA - Hierarchical Modeling Algorithm */
+    ModelType[ModelType["MODEL_TYPE_HMA"] = 9] = "MODEL_TYPE_HMA";
+    ModelType[ModelType["UNRECOGNIZED"] = -1] = "UNRECOGNIZED";
+})(ModelType || (exports.ModelType = ModelType = {}));
+function modelTypeFromJSON(object) {
+    switch (object) {
+        case 0:
+        case "MODEL_TYPE_UNSPECIFIED":
+            return ModelType.MODEL_TYPE_UNSPECIFIED;
+        case 1:
+        case "MODEL_TYPE_CTGAN":
+            return ModelType.MODEL_TYPE_CTGAN;
+        case 2:
+        case "MODEL_TYPE_TVAE":
+            return ModelType.MODEL_TYPE_TVAE;
+        case 3:
+        case "MODEL_TYPE_COPULA":
+            return ModelType.MODEL_TYPE_COPULA;
+        case 4:
+        case "MODEL_TYPE_FAST":
+            return ModelType.MODEL_TYPE_FAST;
+        case 5:
+        case "MODEL_TYPE_GAN":
+            return ModelType.MODEL_TYPE_GAN;
+        case 6:
+        case "MODEL_TYPE_WGAN":
+            return ModelType.MODEL_TYPE_WGAN;
+        case 7:
+        case "MODEL_TYPE_SMOTE":
+            return ModelType.MODEL_TYPE_SMOTE;
+        case 8:
+        case "MODEL_TYPE_BAYESIAN":
+            return ModelType.MODEL_TYPE_BAYESIAN;
+        case 9:
+        case "MODEL_TYPE_HMA":
+            return ModelType.MODEL_TYPE_HMA;
+        case -1:
+        case "UNRECOGNIZED":
+        default:
+            return ModelType.UNRECOGNIZED;
+    }
+}
+function modelTypeToJSON(object) {
+    switch (object) {
+        case ModelType.MODEL_TYPE_UNSPECIFIED:
+            return "MODEL_TYPE_UNSPECIFIED";
+        case ModelType.MODEL_TYPE_CTGAN:
+            return "MODEL_TYPE_CTGAN";
+        case ModelType.MODEL_TYPE_TVAE:
+            return "MODEL_TYPE_TVAE";
+        case ModelType.MODEL_TYPE_COPULA:
+            return "MODEL_TYPE_COPULA";
+        case ModelType.MODEL_TYPE_FAST:
+            return "MODEL_TYPE_FAST";
+        case ModelType.MODEL_TYPE_GAN:
+            return "MODEL_TYPE_GAN";
+        case ModelType.MODEL_TYPE_WGAN:
+            return "MODEL_TYPE_WGAN";
+        case ModelType.MODEL_TYPE_SMOTE:
+            return "MODEL_TYPE_SMOTE";
+        case ModelType.MODEL_TYPE_BAYESIAN:
+            return "MODEL_TYPE_BAYESIAN";
+        case ModelType.MODEL_TYPE_HMA:
+            return "MODEL_TYPE_HMA";
+        case ModelType.UNRECOGNIZED:
+        default:
+            return "UNRECOGNIZED";
+    }
+}
+function createBaseConfiguration() {
+    return { numRecords: 0, dataQuality: undefined, privacy: undefined, model: undefined };
+}
+exports.Configuration = {
+    encode(message, writer = new wire_1.BinaryWriter()) {
+        if (message.numRecords !== 0) {
+            writer.uint32(8).int64(message.numRecords);
+        }
+        if (message.dataQuality !== undefined) {
+            exports.DataQualityConfig.encode(message.dataQuality, writer.uint32(26).fork()).join();
+        }
+        if (message.privacy !== undefined) {
+            exports.PrivacyConfig.encode(message.privacy, writer.uint32(34).fork()).join();
+        }
+        if (message.model !== undefined) {
+            exports.ModelConfig.encode(message.model, writer.uint32(42).fork()).join();
+        }
+        return writer;
+    },
+    decode(input, length) {
+        const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
+        let end = length === undefined ? reader.len : reader.pos + length;
+        const message = createBaseConfiguration();
+        while (reader.pos < end) {
+            const tag = reader.uint32();
+            switch (tag >>> 3) {
+                case 1: {
+                    if (tag !== 8) {
+                        break;
+                    }
+                    message.numRecords = longToNumber(reader.int64());
+                    continue;
+                }
+                case 3: {
+                    if (tag !== 26) {
+                        break;
+                    }
+                    message.dataQuality = exports.DataQualityConfig.decode(reader, reader.uint32());
+                    continue;
+                }
+                case 4: {
+                    if (tag !== 34) {
+                        break;
+                    }
+                    message.privacy = exports.PrivacyConfig.decode(reader, reader.uint32());
+                    continue;
+                }
+                case 5: {
+                    if (tag !== 42) {
+                        break;
+                    }
+                    message.model = exports.ModelConfig.decode(reader, reader.uint32());
+                    continue;
+                }
+            }
+            if ((tag & 7) === 4 || tag === 0) {
+                break;
+            }
+            reader.skip(tag & 7);
+        }
+        return message;
+    },
+    fromJSON(object) {
+        return {
+            numRecords: isSet(object.numRecords) ? globalThis.Number(object.numRecords) : 0,
+            dataQuality: isSet(object.dataQuality) ? exports.DataQualityConfig.fromJSON(object.dataQuality) : undefined,
+            privacy: isSet(object.privacy) ? exports.PrivacyConfig.fromJSON(object.privacy) : undefined,
+            model: isSet(object.model) ? exports.ModelConfig.fromJSON(object.model) : undefined,
+        };
+    },
+    toJSON(message) {
+        const obj = {};
+        if (message.numRecords !== 0) {
+            obj.numRecords = Math.round(message.numRecords);
+        }
+        if (message.dataQuality !== undefined) {
+            obj.dataQuality = exports.DataQualityConfig.toJSON(message.dataQuality);
+        }
+        if (message.privacy !== undefined) {
+            obj.privacy = exports.PrivacyConfig.toJSON(message.privacy);
+        }
+        if (message.model !== undefined) {
+            obj.model = exports.ModelConfig.toJSON(message.model);
+        }
+        return obj;
+    },
+    create(base) {
+        return exports.Configuration.fromPartial(base !== null && base !== void 0 ? base : {});
+    },
+    fromPartial(object) {
+        var _a;
+        const message = createBaseConfiguration();
+        message.numRecords = (_a = object.numRecords) !== null && _a !== void 0 ? _a : 0;
+        message.dataQuality = (object.dataQuality !== undefined && object.dataQuality !== null)
+            ? exports.DataQualityConfig.fromPartial(object.dataQuality)
+            : undefined;
+        message.privacy = (object.privacy !== undefined && object.privacy !== null)
+            ? exports.PrivacyConfig.fromPartial(object.privacy)
+            : undefined;
+        message.model = (object.model !== undefined && object.model !== null)
+            ? exports.ModelConfig.fromPartial(object.model)
+            : undefined;
+        return message;
+    },
+};
+function createBaseDataQualityConfig() {
+    return {
+        correlationAccuracy: 0,
+        statisticalSimilarity: 0,
+        preserveOutliers: false,
+        enforceConstraints: false,
+        constraints: [],
+    };
+}
+exports.DataQualityConfig = {
+    encode(message, writer = new wire_1.BinaryWriter()) {
+        if (message.correlationAccuracy !== 0) {
+            writer.uint32(13).float(message.correlationAccuracy);
+        }
+        if (message.statisticalSimilarity !== 0) {
+            writer.uint32(21).float(message.statisticalSimilarity);
+        }
+        if (message.preserveOutliers !== false) {
+            writer.uint32(24).bool(message.preserveOutliers);
+        }
+        if (message.enforceConstraints !== false) {
+            writer.uint32(32).bool(message.enforceConstraints);
+        }
+        for (const v of message.constraints) {
+            exports.ColumnConstraint.encode(v, writer.uint32(42).fork()).join();
+        }
+        return writer;
+    },
+    decode(input, length) {
+        const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
+        let end = length === undefined ? reader.len : reader.pos + length;
+        const message = createBaseDataQualityConfig();
+        while (reader.pos < end) {
+            const tag = reader.uint32();
+            switch (tag >>> 3) {
+                case 1: {
+                    if (tag !== 13) {
+                        break;
+                    }
+                    message.correlationAccuracy = reader.float();
+                    continue;
+                }
+                case 2: {
+                    if (tag !== 21) {
+                        break;
+                    }
+                    message.statisticalSimilarity = reader.float();
+                    continue;
+                }
+                case 3: {
+                    if (tag !== 24) {
+                        break;
+                    }
+                    message.preserveOutliers = reader.bool();
+                    continue;
+                }
+                case 4: {
+                    if (tag !== 32) {
+                        break;
+                    }
+                    message.enforceConstraints = reader.bool();
+                    continue;
+                }
+                case 5: {
+                    if (tag !== 42) {
+                        break;
+                    }
+                    message.constraints.push(exports.ColumnConstraint.decode(reader, reader.uint32()));
+                    continue;
+                }
+            }
+            if ((tag & 7) === 4 || tag === 0) {
+                break;
+            }
+            reader.skip(tag & 7);
+        }
+        return message;
+    },
+    fromJSON(object) {
+        return {
+            correlationAccuracy: isSet(object.correlationAccuracy) ? globalThis.Number(object.correlationAccuracy) : 0,
+            statisticalSimilarity: isSet(object.statisticalSimilarity) ? globalThis.Number(object.statisticalSimilarity) : 0,
+            preserveOutliers: isSet(object.preserveOutliers) ? globalThis.Boolean(object.preserveOutliers) : false,
+            enforceConstraints: isSet(object.enforceConstraints) ? globalThis.Boolean(object.enforceConstraints) : false,
+            constraints: globalThis.Array.isArray(object === null || object === void 0 ? void 0 : object.constraints)
+                ? object.constraints.map((e) => exports.ColumnConstraint.fromJSON(e))
+                : [],
+        };
+    },
+    toJSON(message) {
+        var _a;
+        const obj = {};
+        if (message.correlationAccuracy !== 0) {
+            obj.correlationAccuracy = message.correlationAccuracy;
+        }
+        if (message.statisticalSimilarity !== 0) {
+            obj.statisticalSimilarity = message.statisticalSimilarity;
+        }
+        if (message.preserveOutliers !== false) {
+            obj.preserveOutliers = message.preserveOutliers;
+        }
+        if (message.enforceConstraints !== false) {
+            obj.enforceConstraints = message.enforceConstraints;
+        }
+        if ((_a = message.constraints) === null || _a === void 0 ? void 0 : _a.length) {
+            obj.constraints = message.constraints.map((e) => exports.ColumnConstraint.toJSON(e));
+        }
+        return obj;
+    },
+    create(base) {
+        return exports.DataQualityConfig.fromPartial(base !== null && base !== void 0 ? base : {});
+    },
+    fromPartial(object) {
+        var _a, _b, _c, _d, _e;
+        const message = createBaseDataQualityConfig();
+        message.correlationAccuracy = (_a = object.correlationAccuracy) !== null && _a !== void 0 ? _a : 0;
+        message.statisticalSimilarity = (_b = object.statisticalSimilarity) !== null && _b !== void 0 ? _b : 0;
+        message.preserveOutliers = (_c = object.preserveOutliers) !== null && _c !== void 0 ? _c : false;
+        message.enforceConstraints = (_d = object.enforceConstraints) !== null && _d !== void 0 ? _d : false;
+        message.constraints = ((_e = object.constraints) === null || _e === void 0 ? void 0 : _e.map((e) => exports.ColumnConstraint.fromPartial(e))) || [];
+        return message;
+    },
+};
+function createBaseColumnConstraint() {
+    return { columnName: "", constraintType: "", minValue: "", maxValue: "", regexPattern: "", allowedValues: [] };
+}
+exports.ColumnConstraint = {
+    encode(message, writer = new wire_1.BinaryWriter()) {
+        if (message.columnName !== "") {
+            writer.uint32(10).string(message.columnName);
+        }
+        if (message.constraintType !== "") {
+            writer.uint32(18).string(message.constraintType);
+        }
+        if (message.minValue !== "") {
+            writer.uint32(26).string(message.minValue);
+        }
+        if (message.maxValue !== "") {
+            writer.uint32(34).string(message.maxValue);
+        }
+        if (message.regexPattern !== "") {
+            writer.uint32(42).string(message.regexPattern);
+        }
+        for (const v of message.allowedValues) {
+            writer.uint32(50).string(v);
+        }
+        return writer;
+    },
+    decode(input, length) {
+        const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
+        let end = length === undefined ? reader.len : reader.pos + length;
+        const message = createBaseColumnConstraint();
+        while (reader.pos < end) {
+            const tag = reader.uint32();
+            switch (tag >>> 3) {
+                case 1: {
+                    if (tag !== 10) {
+                        break;
+                    }
+                    message.columnName = reader.string();
+                    continue;
+                }
+                case 2: {
+                    if (tag !== 18) {
+                        break;
+                    }
+                    message.constraintType = reader.string();
+                    continue;
+                }
+                case 3: {
+                    if (tag !== 26) {
+                        break;
+                    }
+                    message.minValue = reader.string();
+                    continue;
+                }
+                case 4: {
+                    if (tag !== 34) {
+                        break;
+                    }
+                    message.maxValue = reader.string();
+                    continue;
+                }
+                case 5: {
+                    if (tag !== 42) {
+                        break;
+                    }
+                    message.regexPattern = reader.string();
+                    continue;
+                }
+                case 6: {
+                    if (tag !== 50) {
+                        break;
+                    }
+                    message.allowedValues.push(reader.string());
+                    continue;
+                }
+            }
+            if ((tag & 7) === 4 || tag === 0) {
+                break;
+            }
+            reader.skip(tag & 7);
+        }
+        return message;
+    },
+    fromJSON(object) {
+        return {
+            columnName: isSet(object.columnName) ? globalThis.String(object.columnName) : "",
+            constraintType: isSet(object.constraintType) ? globalThis.String(object.constraintType) : "",
+            minValue: isSet(object.minValue) ? globalThis.String(object.minValue) : "",
+            maxValue: isSet(object.maxValue) ? globalThis.String(object.maxValue) : "",
+            regexPattern: isSet(object.regexPattern) ? globalThis.String(object.regexPattern) : "",
+            allowedValues: globalThis.Array.isArray(object === null || object === void 0 ? void 0 : object.allowedValues)
+                ? object.allowedValues.map((e) => globalThis.String(e))
+                : [],
+        };
+    },
+    toJSON(message) {
+        var _a;
+        const obj = {};
+        if (message.columnName !== "") {
+            obj.columnName = message.columnName;
+        }
+        if (message.constraintType !== "") {
+            obj.constraintType = message.constraintType;
+        }
+        if (message.minValue !== "") {
+            obj.minValue = message.minValue;
+        }
+        if (message.maxValue !== "") {
+            obj.maxValue = message.maxValue;
+        }
+        if (message.regexPattern !== "") {
+            obj.regexPattern = message.regexPattern;
+        }
+        if ((_a = message.allowedValues) === null || _a === void 0 ? void 0 : _a.length) {
+            obj.allowedValues = message.allowedValues;
+        }
+        return obj;
+    },
+    create(base) {
+        return exports.ColumnConstraint.fromPartial(base !== null && base !== void 0 ? base : {});
+    },
+    fromPartial(object) {
+        var _a, _b, _c, _d, _e, _f;
+        const message = createBaseColumnConstraint();
+        message.columnName = (_a = object.columnName) !== null && _a !== void 0 ? _a : "";
+        message.constraintType = (_b = object.constraintType) !== null && _b !== void 0 ? _b : "";
+        message.minValue = (_c = object.minValue) !== null && _c !== void 0 ? _c : "";
+        message.maxValue = (_d = object.maxValue) !== null && _d !== void 0 ? _d : "";
+        message.regexPattern = (_e = object.regexPattern) !== null && _e !== void 0 ? _e : "";
+        message.allowedValues = ((_f = object.allowedValues) === null || _f === void 0 ? void 0 : _f.map((e) => e)) || [];
+        return message;
+    },
+};
+function createBasePrivacyConfig() {
+    return { epsilon: 0, kAnonymityCheck: false, kValue: 0, sensitiveColumns: [], enablePrivacyMetrics: false };
+}
+exports.PrivacyConfig = {
+    encode(message, writer = new wire_1.BinaryWriter()) {
+        if (message.epsilon !== 0) {
+            writer.uint32(13).float(message.epsilon);
+        }
+        if (message.kAnonymityCheck !== false) {
+            writer.uint32(16).bool(message.kAnonymityCheck);
+        }
+        if (message.kValue !== 0) {
+            writer.uint32(24).int32(message.kValue);
+        }
+        for (const v of message.sensitiveColumns) {
+            writer.uint32(34).string(v);
+        }
+        if (message.enablePrivacyMetrics !== false) {
+            writer.uint32(40).bool(message.enablePrivacyMetrics);
+        }
+        return writer;
+    },
+    decode(input, length) {
+        const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
+        let end = length === undefined ? reader.len : reader.pos + length;
+        const message = createBasePrivacyConfig();
+        while (reader.pos < end) {
+            const tag = reader.uint32();
+            switch (tag >>> 3) {
+                case 1: {
+                    if (tag !== 13) {
+                        break;
+                    }
+                    message.epsilon = reader.float();
+                    continue;
+                }
+                case 2: {
+                    if (tag !== 16) {
+                        break;
+                    }
+                    message.kAnonymityCheck = reader.bool();
+                    continue;
+                }
+                case 3: {
+                    if (tag !== 24) {
+                        break;
+                    }
+                    message.kValue = reader.int32();
+                    continue;
+                }
+                case 4: {
+                    if (tag !== 34) {
+                        break;
+                    }
+                    message.sensitiveColumns.push(reader.string());
+                    continue;
+                }
+                case 5: {
+                    if (tag !== 40) {
+                        break;
+                    }
+                    message.enablePrivacyMetrics = reader.bool();
+                    continue;
+                }
+            }
+            if ((tag & 7) === 4 || tag === 0) {
+                break;
+            }
+            reader.skip(tag & 7);
+        }
+        return message;
+    },
+    fromJSON(object) {
+        return {
+            epsilon: isSet(object.epsilon) ? globalThis.Number(object.epsilon) : 0,
+            kAnonymityCheck: isSet(object.kAnonymityCheck) ? globalThis.Boolean(object.kAnonymityCheck) : false,
+            kValue: isSet(object.kValue) ? globalThis.Number(object.kValue) : 0,
+            sensitiveColumns: globalThis.Array.isArray(object === null || object === void 0 ? void 0 : object.sensitiveColumns)
+                ? object.sensitiveColumns.map((e) => globalThis.String(e))
+                : [],
+            enablePrivacyMetrics: isSet(object.enablePrivacyMetrics)
+                ? globalThis.Boolean(object.enablePrivacyMetrics)
+                : false,
+        };
+    },
+    toJSON(message) {
+        var _a;
+        const obj = {};
+        if (message.epsilon !== 0) {
+            obj.epsilon = message.epsilon;
+        }
+        if (message.kAnonymityCheck !== false) {
+            obj.kAnonymityCheck = message.kAnonymityCheck;
+        }
+        if (message.kValue !== 0) {
+            obj.kValue = Math.round(message.kValue);
+        }
+        if ((_a = message.sensitiveColumns) === null || _a === void 0 ? void 0 : _a.length) {
+            obj.sensitiveColumns = message.sensitiveColumns;
+        }
+        if (message.enablePrivacyMetrics !== false) {
+            obj.enablePrivacyMetrics = message.enablePrivacyMetrics;
+        }
+        return obj;
+    },
+    create(base) {
+        return exports.PrivacyConfig.fromPartial(base !== null && base !== void 0 ? base : {});
+    },
+    fromPartial(object) {
+        var _a, _b, _c, _d, _e;
+        const message = createBasePrivacyConfig();
+        message.epsilon = (_a = object.epsilon) !== null && _a !== void 0 ? _a : 0;
+        message.kAnonymityCheck = (_b = object.kAnonymityCheck) !== null && _b !== void 0 ? _b : false;
+        message.kValue = (_c = object.kValue) !== null && _c !== void 0 ? _c : 0;
+        message.sensitiveColumns = ((_d = object.sensitiveColumns) === null || _d === void 0 ? void 0 : _d.map((e) => e)) || [];
+        message.enablePrivacyMetrics = (_e = object.enablePrivacyMetrics) !== null && _e !== void 0 ? _e : false;
+        return message;
+    },
+};
+function createBaseModelConfig() {
+    return { modelType: 0, epochs: 0, learningRate: 0, batchSize: 0, customParameters: {} };
+}
+exports.ModelConfig = {
+    encode(message, writer = new wire_1.BinaryWriter()) {
+        if (message.modelType !== 0) {
+            writer.uint32(8).int32(message.modelType);
+        }
+        if (message.epochs !== 0) {
+            writer.uint32(16).int32(message.epochs);
+        }
+        if (message.learningRate !== 0) {
+            writer.uint32(29).float(message.learningRate);
+        }
+        if (message.batchSize !== 0) {
+            writer.uint32(32).int32(message.batchSize);
+        }
+        Object.entries(message.customParameters).forEach(([key, value]) => {
+            exports.ModelConfig_CustomParametersEntry.encode({ key: key, value }, writer.uint32(42).fork()).join();
+        });
+        return writer;
+    },
+    decode(input, length) {
+        const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
+        let end = length === undefined ? reader.len : reader.pos + length;
+        const message = createBaseModelConfig();
+        while (reader.pos < end) {
+            const tag = reader.uint32();
+            switch (tag >>> 3) {
+                case 1: {
+                    if (tag !== 8) {
+                        break;
+                    }
+                    message.modelType = reader.int32();
+                    continue;
+                }
+                case 2: {
+                    if (tag !== 16) {
+                        break;
+                    }
+                    message.epochs = reader.int32();
+                    continue;
+                }
+                case 3: {
+                    if (tag !== 29) {
+                        break;
+                    }
+                    message.learningRate = reader.float();
+                    continue;
+                }
+                case 4: {
+                    if (tag !== 32) {
+                        break;
+                    }
+                    message.batchSize = reader.int32();
+                    continue;
+                }
+                case 5: {
+                    if (tag !== 42) {
+                        break;
+                    }
+                    const entry5 = exports.ModelConfig_CustomParametersEntry.decode(reader, reader.uint32());
+                    if (entry5.value !== undefined) {
+                        message.customParameters[entry5.key] = entry5.value;
+                    }
+                    continue;
+                }
+            }
+            if ((tag & 7) === 4 || tag === 0) {
+                break;
+            }
+            reader.skip(tag & 7);
+        }
+        return message;
+    },
+    fromJSON(object) {
+        return {
+            modelType: isSet(object.modelType) ? modelTypeFromJSON(object.modelType) : 0,
+            epochs: isSet(object.epochs) ? globalThis.Number(object.epochs) : 0,
+            learningRate: isSet(object.learningRate) ? globalThis.Number(object.learningRate) : 0,
+            batchSize: isSet(object.batchSize) ? globalThis.Number(object.batchSize) : 0,
+            customParameters: isObject(object.customParameters)
+                ? Object.entries(object.customParameters).reduce((acc, [key, value]) => {
+                    acc[key] = String(value);
+                    return acc;
+                }, {})
+                : {},
+        };
+    },
+    toJSON(message) {
+        const obj = {};
+        if (message.modelType !== 0) {
+            obj.modelType = modelTypeToJSON(message.modelType);
+        }
+        if (message.epochs !== 0) {
+            obj.epochs = Math.round(message.epochs);
+        }
+        if (message.learningRate !== 0) {
+            obj.learningRate = message.learningRate;
+        }
+        if (message.batchSize !== 0) {
+            obj.batchSize = Math.round(message.batchSize);
+        }
+        if (message.customParameters) {
+            const entries = Object.entries(message.customParameters);
+            if (entries.length > 0) {
+                obj.customParameters = {};
+                entries.forEach(([k, v]) => {
+                    obj.customParameters[k] = v;
+                });
+            }
+        }
+        return obj;
+    },
+    create(base) {
+        return exports.ModelConfig.fromPartial(base !== null && base !== void 0 ? base : {});
+    },
+    fromPartial(object) {
+        var _a, _b, _c, _d, _e;
+        const message = createBaseModelConfig();
+        message.modelType = (_a = object.modelType) !== null && _a !== void 0 ? _a : 0;
+        message.epochs = (_b = object.epochs) !== null && _b !== void 0 ? _b : 0;
+        message.learningRate = (_c = object.learningRate) !== null && _c !== void 0 ? _c : 0;
+        message.batchSize = (_d = object.batchSize) !== null && _d !== void 0 ? _d : 0;
+        message.customParameters = Object.entries((_e = object.customParameters) !== null && _e !== void 0 ? _e : {}).reduce((acc, [key, value]) => {
+            if (value !== undefined) {
+                acc[key] = globalThis.String(value);
+            }
+            return acc;
+        }, {});
+        return message;
+    },
+};
+function createBaseModelConfig_CustomParametersEntry() {
+    return { key: "", value: "" };
+}
+exports.ModelConfig_CustomParametersEntry = {
+    encode(message, writer = new wire_1.BinaryWriter()) {
+        if (message.key !== "") {
+            writer.uint32(10).string(message.key);
+        }
+        if (message.value !== "") {
+            writer.uint32(18).string(message.value);
+        }
+        return writer;
+    },
+    decode(input, length) {
+        const reader = input instanceof wire_1.BinaryReader ? input : new wire_1.BinaryReader(input);
+        let end = length === undefined ? reader.len : reader.pos + length;
+        const message = createBaseModelConfig_CustomParametersEntry();
+        while (reader.pos < end) {
+            const tag = reader.uint32();
+            switch (tag >>> 3) {
+                case 1: {
+                    if (tag !== 10) {
+                        break;
+                    }
+                    message.key = reader.string();
+                    continue;
+                }
+                case 2: {
+                    if (tag !== 18) {
+                        break;
+                    }
+                    message.value = reader.string();
+                    continue;
+                }
+            }
+            if ((tag & 7) === 4 || tag === 0) {
+                break;
+            }
+            reader.skip(tag & 7);
+        }
+        return message;
+    },
+    fromJSON(object) {
+        return {
+            key: isSet(object.key) ? globalThis.String(object.key) : "",
+            value: isSet(object.value) ? globalThis.String(object.value) : "",
+        };
+    },
+    toJSON(message) {
+        const obj = {};
+        if (message.key !== "") {
+            obj.key = message.key;
+        }
+        if (message.value !== "") {
+            obj.value = message.value;
+        }
+        return obj;
+    },
+    create(base) {
+        return exports.ModelConfig_CustomParametersEntry.fromPartial(base !== null && base !== void 0 ? base : {});
+    },
+    fromPartial(object) {
+        var _a, _b;
+        const message = createBaseModelConfig_CustomParametersEntry();
+        message.key = (_a = object.key) !== null && _a !== void 0 ? _a : "";
+        message.value = (_b = object.value) !== null && _b !== void 0 ? _b : "";
+        return message;
+    },
+};
+function longToNumber(int64) {
+    const num = globalThis.Number(int64.toString());
+    if (num > globalThis.Number.MAX_SAFE_INTEGER) {
+        throw new globalThis.Error("Value is larger than Number.MAX_SAFE_INTEGER");
+    }
+    if (num < globalThis.Number.MIN_SAFE_INTEGER) {
+        throw new globalThis.Error("Value is smaller than Number.MIN_SAFE_INTEGER");
+    }
+    return num;
+}
+function isObject(value) {
+    return typeof value === "object" && value !== null;
+}
+function isSet(value) {
+    return value !== null && value !== undefined;
+}