パブリック最終クラスBoostedTreesTrainingPredict
入力インスタンスに対して複数の加法回帰アンサンブル予測子を実行し、
キャッシュされたロジットの更新を計算します。トレーニング中に使用することを目的として設計されています。キャッシュされたツリー ID とキャッシュされたノード ID から開始してツリーを走査し、キャッシュにプッシュされる更新を計算します。
パブリックメソッド
静的BoostedTreesTrainingPredict | |
出力<整数> | ノードID () 新しい Tree_id に新しいノード ID を含むランク 1 テンソル。 |
出力<浮動小数点数> | 部分ロジッツ() 各例のロジット更新 (保存されたキャッシュされた値に関して) を含むランク 2 テンソル。 |
出力<整数> | ツリーID () 各例の新しいツリー ID を含むランク 1 テンソル。 |
継承されたメソッド
パブリックメソッド
public static BoostedTreesTrainingPredict create (スコープscope、オペランド<?>treeEnsembleHandle、オペランド<Integer>cachedTreeIds、オペランド<Integer>cachedNodeIds、Iterable<オペランド<Integer>>bucketized features、Long logitsDimension)
新しい BoostedTreesTrainingPredict オペレーションをラップするクラスを作成するためのファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
キャッシュされたツリー ID | 予測の開始ツリーであるキャッシュされたツリー ID を含むランク 1 テンソル。 |
キャッシュされたノードID | 予測の開始ノードであるキャッシュされたノード ID を含むランク 1 テンソル。 |
バケット化された機能 | 各特徴のバケット ID を含むランク 1 Tensor のリスト。 |
ロジッツ寸法 | スカラー、ロジットの次元。部分的なロジット形状に使用されます。 |
戻り値
- BoostedTreesTrainingPredict の新しいインスタンス