mirror of
https://github.com/microsoft/autogen.git
synced 2025-09-11 17:26:02 +00:00
98 lines
2.1 KiB
Python
98 lines
2.1 KiB
Python
|
ParamList_LightGBM_Base = [
|
||
|
|
"baggingFraction",
|
||
|
|
"baggingFreq",
|
||
|
|
"baggingSeed",
|
||
|
|
"binSampleCount",
|
||
|
|
"boostFromAverage",
|
||
|
|
"boostingType",
|
||
|
|
"catSmooth",
|
||
|
|
"categoricalSlotIndexes",
|
||
|
|
"categoricalSlotNames",
|
||
|
|
"catl2",
|
||
|
|
"chunkSize",
|
||
|
|
"dataRandomSeed",
|
||
|
|
"defaultListenPort",
|
||
|
|
"deterministic",
|
||
|
|
"driverListenPort",
|
||
|
|
"dropRate",
|
||
|
|
"dropSeed",
|
||
|
|
"earlyStoppingRound",
|
||
|
|
"executionMode",
|
||
|
|
"extraSeed" "featureFraction",
|
||
|
|
"featureFractionByNode",
|
||
|
|
"featureFractionSeed",
|
||
|
|
"featuresCol",
|
||
|
|
"featuresShapCol",
|
||
|
|
"fobj" "improvementTolerance",
|
||
|
|
"initScoreCol",
|
||
|
|
"isEnableSparse",
|
||
|
|
"isProvideTrainingMetric",
|
||
|
|
"labelCol",
|
||
|
|
"lambdaL1",
|
||
|
|
"lambdaL2",
|
||
|
|
"leafPredictionCol",
|
||
|
|
"learningRate",
|
||
|
|
"matrixType",
|
||
|
|
"maxBin",
|
||
|
|
"maxBinByFeature",
|
||
|
|
"maxCatThreshold",
|
||
|
|
"maxCatToOnehot",
|
||
|
|
"maxDeltaStep",
|
||
|
|
"maxDepth",
|
||
|
|
"maxDrop",
|
||
|
|
"metric",
|
||
|
|
"microBatchSize",
|
||
|
|
"minDataInLeaf",
|
||
|
|
"minDataPerBin",
|
||
|
|
"minDataPerGroup",
|
||
|
|
"minGainToSplit",
|
||
|
|
"minSumHessianInLeaf",
|
||
|
|
"modelString",
|
||
|
|
"monotoneConstraints",
|
||
|
|
"monotoneConstraintsMethod",
|
||
|
|
"monotonePenalty",
|
||
|
|
"negBaggingFraction",
|
||
|
|
"numBatches",
|
||
|
|
"numIterations",
|
||
|
|
"numLeaves",
|
||
|
|
"numTasks",
|
||
|
|
"numThreads",
|
||
|
|
"objectiveSeed",
|
||
|
|
"otherRate",
|
||
|
|
"parallelism",
|
||
|
|
"passThroughArgs",
|
||
|
|
"posBaggingFraction",
|
||
|
|
"predictDisableShapeCheck",
|
||
|
|
"predictionCol",
|
||
|
|
"repartitionByGroupingColumn",
|
||
|
|
"seed",
|
||
|
|
"skipDrop",
|
||
|
|
"slotNames",
|
||
|
|
"timeout",
|
||
|
|
"topK",
|
||
|
|
"topRate",
|
||
|
|
"uniformDrop",
|
||
|
|
"useBarrierExecutionMode",
|
||
|
|
"useMissing",
|
||
|
|
"useSingleDatasetMode",
|
||
|
|
"validationIndicatorCol",
|
||
|
|
"verbosity",
|
||
|
|
"weightCol",
|
||
|
|
"xGBoostDartMode",
|
||
|
|
"zeroAsMissing",
|
||
|
|
"objective",
|
||
|
|
]
|
||
|
|
ParamList_LightGBM_Classifier = ParamList_LightGBM_Base + [
|
||
|
|
"isUnbalance",
|
||
|
|
"probabilityCol",
|
||
|
|
"rawPredictionCol",
|
||
|
|
"thresholds",
|
||
|
|
]
|
||
|
|
ParamList_LightGBM_Regressor = ParamList_LightGBM_Base + ["tweedieVariancePower"]
|
||
|
|
ParamList_LightGBM_Ranker = ParamList_LightGBM_Base + [
|
||
|
|
"groupCol",
|
||
|
|
"evalAt",
|
||
|
|
"labelGain",
|
||
|
|
"maxPosition",
|
||
|
|
]
|