ブーストツリートレーニング予測

パブリック最終クラスBoostedTreesTrainingPredict

入力インスタンスに対して複数の加法回帰アンサンブル予測子を実行し、

キャッシュされたロジットの更新を計算します。トレーニング中に使用することを目的として設計されています。キャッシュされたツリー ID とキャッシュされたノード ID から開始してツリーを走査し、キャッシュにプッシュされる更新を計算します。

パブリックメソッド

静的BoostedTreesTrainingPredict
create (スコープscope、オペランド<?>treeEnsembleHandle、オペランド<Integer>cachedTreeIds、オペランド<Integer>cachedNodeIds、Iterable<オペランド<Integer>>bucketized features、Long logitsDimension)
新しい 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 の新しいインスタンス

public Output <Integer>ノード ID ()

新しい Tree_id に新しいノード ID を含むランク 1 テンソル。

public出力<Float>部分Logits ()

各例のロジット更新 (保存されたキャッシュされた値に関して) を含むランク 2 テンソル。

public Output <Integer>ツリー ID ()

各例の新しいツリー ID を含むランク 1 テンソル。