BoostedTreesSparseCalculateBestFeatureSplit

パブリック最終クラスBoostedTreesSparseCalculateBestFeatureSplit

各特徴のゲインを計算し、その特徴に対して可能な限り最適な分割情報を返します。

分割情報は、各特徴のノードごとの最適なしきい値 (バケット ID)、ゲイン、および左右のノードの寄与です。

すべてのノードを各フィーチャで分割できない可能性があります。したがって、可能なノードのリストはフィーチャ間で異なる場合があります。したがって、この機能を分割するために使用できるノードのリストを含む、各機能の「node_ids_list」を返します。

この方法では、出力は機能ごとおよびノー​​ドごとに最適な分割となるため、(可能なすべての機能の中で) 各ノードに最適な分割を生成するには、後で結合する必要があります。

出力形状は、すべてのテンソルの最初の次元が同じであり、各フィーチャの可能な分割ノードの数に等しいという点で互換性があります。

ネストされたクラス

クラスBoostedTreesSparseCalculateBestFeatureSplit.Options BoostedTreesSparseCalculateBestFeatureSplitのオプション属性

パブリックメソッド

静的BoostedTreesSparseCalculateBestFeatureSplit
create ( Scopeスコープ、オペランド<Integer> nodeIdRange、オペランド<Integer> statssummaryIndi​​ces、オペランド<Float> statssummaryValues、オペランド<Integer> statssummaryShape、オペランド<Float> l1、オペランド<Float> l2、オペランド<Float> TreeComplexity、オペランド<Float > minNodeWeight、Long logitsDimension、オプション...オプション)
新しい BoostedTreesSparseCalculateBestFeatureSplit オペレーションをラップするクラスを作成するファクトリ メソッド。
出力<整数>
機能寸法()
各ノードに分割する各フィーチャの最適なフィーチャ ディメンションを示すランク 1 テンソル。
出力<浮動小数点数>
利益()
各ノードを分割するための最適なゲインを示すランク 1 テンソル。
出力<浮動小数点数>
leftNodeContribs ()
各特徴の指定されたしきい値によって親ノードから左方向に分岐するときの左ノードの寄与を示すランク 2 テンソル。
出力<整数>
ノードID ()
分割可能なノード ID を示すランク 1 テンソル。
出力<浮動小数点数>
rightNodeContribs ()
ランク 2 テンソル。 left_node_contribs_list と同じ形状/条件を持ちますが、値が右側のノードのものであるだけです。
静的BoostedTreesSparseCalculateBestFeatureSplit.Options
splitType (文字列分割タイプ)
出力<文字列>
splitWithDefaultDirections ()
データが欠落している場合にどの方向に進むかを示すランク 1 テンソル。
出力<整数>
しきい値()
各ノードでの分割に関して (しきい値として) 比較するバケット ID を示すランク 1 テンソル。

継承されたメソッド

パブリックメソッド

public static BoostedTreesSparseCalculateBestFeatureSplit create ( Scopeスコープ、オペランド<Integer> nodeIdRange、オペランド<Integer> statssummaryIndi​​ces、オペランド<Float> statssummaryValues、オペランド<Integer> statssummaryShape、オペランド<Float> l1、オペランド<Float> l2、オペランド<Float> TreeComplexity、オペランド<Float> minNodeWeight、Long logitsDimension、 Options...オプション)

新しい BoostedTreesSparseCalculateBestFeatureSplit オペレーションをラップするクラスを作成するファクトリ メソッド。

パラメーター
範囲現在のスコープ
ノードID範囲`stats_summary_list` 内で処理するノード ID の範囲 [first, last) を指定するランク 1 テンソル (shape=[2])。ノードは、「for node_id in range(node_id_range[0], node_id_range[1])」のように、テンソルで指定された 2 つのノード間で反復されます (最後のインデックス node_id_range[1] は排他的であることに注意してください)。
統計概要インデックス各特徴のバケットごとのノードごとの累積統計サマリー (勾配/ヘシアン) の密な形状 [N, 4] (N はゼロ以外の値の数を指定します) のランク 2 int64 テンソル。 2 番目のディメンションには、ノード ID、機能ディメンション、バケット ID、および統計ディメンションが含まれます。 stats dim はロジット次元とヘシアン次元の合計です。ヘシアン次元は、対角ヘシアンが使用されている場合はロジット次元、完全ヘシアンが使用されている場合はロジット次元 ^ 2 のいずれかになります。
統計概要値密な形状 [N] (N はゼロ以外の値の数を指定します) のランク 1 の浮動小数点テンソル。summary_indices の各要素の値を提供します。
統計概要形状密な形状のランク 1 浮動小数点テンソル [4]。スパース テンソルの密な形状を指定します。これは、[num ツリー ノード、フィーチャ ディメンション、数値バケット、統計ディム] です。
l1リーフの重みに関する l1 正則化係数 (インスタンスごと)。
l2リーフの重みに関する l2 正則化係数 (インスタンスごと)。
ツリーの複雑さリーフごとのゲインの調整。
minNodeWeightノードが分割対象とみなされるために必要となる、ノード内のヘシアンの最小平均値。
ロジッツ寸法ロジットの次元、つまりクラスの数。
オプションオプションの属性値を持ちます
戻り値
  • BoostedTreesSparseCalculateBestFeatureSplit の新しいインスタンス

public Output <Integer> featureDimensions ()

各ノードに分割する各フィーチャの最適なフィーチャ ディメンションを示すランク 1 テンソル。

public出力<Float> が増加します()

各ノードを分割するための最適なゲインを示すランク 1 テンソル。

public出力<Float> leftNodeContribs ()

各特徴の指定されたしきい値によって親ノードから左方向に分岐するときの左ノードの寄与を示すランク 2 テンソル。この値は、親ノードの値に加算して左ノードの値を作成するために使用されます。 2 番目の次元のサイズはロジット次元です。

public Output <Integer>ノード ID ()

分割可能なノード ID を示すランク 1 テンソル。

public出力<Float> rightNodeContribs ()

ランク 2 テンソル。 left_node_contribs_list と同じ形状/条件を持ちますが、値が右側のノードのものであるだけです。

public static BoostedTreesSparseCalculateBestFeatureSplit.Options SplitType (String SplitType)

パラメーター
分割タイプこの Op が不等分割または等価分割を実行するかどうかを示す文字列。

public Output <String> splitWithDefaultDirections ()

データが欠落している場合にどの方向に進むかを示すランク 1 テンソル。デフォルトの左との不等号は 0 を返し、デフォルトの右との不等号は 1 を返し、デフォルトの右との等号は 2 を返します。

public出力<Integer>しきい値()

各ノードでの分割に関して (しきい値として) 比較するバケット ID を示すランク 1 テンソル。