ink.flink-ml-benchmark-1.15.2.3.0.source-code.linearregression-benchmark.json Maven / Gradle / Ivy

Go to download
// Licensed to the Apache Software Foundation (ASF) under one or more
// contributor license agreements.  See the NOTICE file distributed with
// this work for additional information regarding copyright ownership.
// The ASF licenses this file to You under the Apache License, Version 2.0
// (the "License"); you may not use this file except in compliance with
// the License.  You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

{
  "version": 1,
  "linearregression": {
    "stage": {
      "className": "org.apache.flink.ml.regression.linearregression.LinearRegression",
      "paramMap": {
        "maxIter": 20,
        "reg": 0.0,
        "elasticNet": 0.0,
        "learningRate": 0.1,
        "globalBatchSize": 100000,
        "tol": 0.000001
      }
    },
    "inputData": {
      "className": "org.apache.flink.ml.benchmark.datagenerator.common.LabeledPointWithWeightGenerator",
      "paramMap": {
        "colNames": [["features", "label", "weight"]],
        "featureArity": 0,
        "labelArity": 10,
        "numValues": 10000000,
        "vectorDim": 100
      }
    }
  }
}