語言

LightGbmRegressionTrainer 類別

定義

用於 IEstimator<TTransformer> 使用 LightGBM 訓練增強決策樹迴歸模型。

public sealed class LightGbmRegressionTrainer : Microsoft.ML.Trainers.LightGbm.LightGbmTrainerBase<Microsoft.ML.Trainers.LightGbm.LightGbmRegressionTrainer.Options,float,Microsoft.ML.Data.RegressionPredictionTransformer<Microsoft.ML.Trainers.LightGbm.LightGbmRegressionModelParameters>,Microsoft.ML.Trainers.LightGbm.LightGbmRegressionModelParameters>
type LightGbmRegressionTrainer = class
    inherit LightGbmTrainerBase<LightGbmRegressionTrainer.Options, single, RegressionPredictionTransformer<LightGbmRegressionModelParameters>, LightGbmRegressionModelParameters>
Public NotInheritable Class LightGbmRegressionTrainer
Inherits LightGbmTrainerBase(Of LightGbmRegressionTrainer.Options, Single, RegressionPredictionTransformer(Of LightGbmRegressionModelParameters), LightGbmRegressionModelParameters)
繼承

備註

要建立這個訓練器,請使用 LightGbm 或 LightGbm(Options)。

輸入與輸出欄位

輸入標籤欄位資料必須是 Single。 輸入特徵欄位資料必須是已知大小的向量。Single

此訓練器會輸出下列欄位:

輸出數據行名稱 欄類型 Description
Score Single 模型預測的無界分數。

訓練師特性

機器學習任務 Regression
需要正規化嗎? No
快取是必須的嗎? No
除了 Microsoft.ML 之外,必須使用 NuGet Microsoft.ML.LightGbm
可匯出至 ONNX Yes

訓練演算法細節

LightGBM 是一個開放原始碼的梯度提升決策樹實作。 有關實作細節,請參閱 LightGBM 官方文件 或此 論文。

請參考「參見」部分,裡面有相關用法的連結。

欄位

名稱 Description
FeatureColumn

訓練師期待的特色欄。

(繼承來源 TrainerEstimatorBase<TTransformer,TModel>)
GroupIdColumn

排名訓練師期望的可選群組ID欄位。

(繼承來源 TrainerEstimatorBaseWithGroupId<TTransformer,TModel>)
LabelColumn

訓練師期望的標籤欄。 可以是 null,表示該標籤不用於訓練。

(繼承來源 TrainerEstimatorBase<TTransformer,TModel>)
WeightColumn

訓練師預期的體重欄。 可以是 null,表示訓練時不使用重量。

(繼承來源 TrainerEstimatorBase<TTransformer,TModel>)

屬性

名稱 Description
Info

用於 IEstimator<TTransformer> 使用 LightGBM 訓練增強決策樹迴歸模型。

(繼承來源 LightGbmTrainerBase<TOptions,TOutput,TTransformer,TModel>)

方法

名稱 Description
Fit(IDataView, IDataView)

使用訓練與驗證資料訓練 a LightGbmRegressionTrainer ,回傳 RegressionPredictionTransformer<TModel>。

Fit(IDataView)

訓練並返回 ITransformer。

(繼承來源 TrainerEstimatorBase<TTransformer,TModel>)
GetOutputSchema(SchemaShape)

用於 IEstimator<TTransformer> 使用 LightGBM 訓練增強決策樹迴歸模型。

(繼承來源 TrainerEstimatorBase<TTransformer,TModel>)

擴充方法

名稱 Description
AppendCacheCheckpoint<TTrans>(IEstimator<TTrans>, IHostEnvironment)

在估計鏈中附加一個「快取檢查點」。 這將確保下游估計器能針對快取資料進行訓練。 在訓練師接受多次資料通行前設置快取檢查點會很有幫助。

WithOnFitDelegate<TTransformer>(IEstimator<TTransformer>, Action<TTransformer>)

給定一個估計器,回傳一個包裹物件,該物件會呼叫一次 Fit(IDataView) 代理。 估計器通常回傳擬合的資訊很重要,因此該 Fit(IDataView) 方法回傳一個特定型別的物件,而非一般 ITransformer的 。 然而,同時, IEstimator<TTransformer> 通常會被組成包含許多物件的管線,因此我們可能需要建立一條估計鏈,將 EstimatorChain<TLastTransformer> 我們想要取得變壓器的估計器埋藏在這條鏈的某處。 在這種情況下,我們可以透過此方法附加一個代理,當 fit 被呼叫時會被呼叫。

適用於

另請參閱