מחשב רווחים עבור כל תכונה ומחזיר את המידע המפוצל הטוב ביותר עבור התכונה.
המידע המפוצל הוא הסף הטוב ביותר (מזהה דלי), רווחים ותרומת צומת שמאל/ימין לכל צומת עבור כל תכונה.
ייתכן שלא ניתן לפצל את כל הצמתים בכל תכונה. לפיכך, רשימת הצמתים האפשריים יכולה להיות שונה בין התכונות. לכן, אנו מחזירים `node_ids_list` עבור כל תכונה, המכילה את רשימת הצמתים שניתן להשתמש בתכונה זו לפיצול.
באופן זה, הפלט הוא הפיצול הטוב ביותר לכל תכונות ולכל צומת, כך שצריך לשלב אותו מאוחר יותר כדי לייצר את הפיצול הטוב ביותר עבור כל צומת (בין כל התכונות האפשריות).
האורך של רשימות הפלט כולן באותו אורך, `num_features`. צורות הפלט תואמות באופן שהמימד הראשון של כל הטנזורים של כל הרשימות זהה ושווה למספר הצמתים המפוצלים האפשריים עבור כל תכונה.
שיטות ציבוריות
| סטטי BoostedTreesCalculateBestGainsPerFeature | |
| רשימה< פלט <צף>> | רשימת רווחים () רשימת פלט של טנסור דרגה 1 המציינת את הרווחים הטובים ביותר עבור כל תכונה לפיצול עבור צמתים מסוימים. |
| רשימה< פלט <צף>> | leftNodeContribsList () רשימה של טנסור דרגה 2 המציינת את תרומתם של הצמתים השמאליים בעת הסתעפות מצמתי אב (ניתן על ידי אלמנט הטנזור ב-output node_ids_list) לכיוון השמאלי לפי הסף הנתון עבור כל תכונה. |
| רשימה< פלט <מספר שלם>> | nodeIdsList () רשימת פלט של טנסור דרגה 1 המציינת מזהי צומת מפוצלים אפשריים עבור כל תכונה. |
| רשימה< פלט <צף>> | rightNodeContribsList () רשימה של טנסור דרגה 2, עם אותם צורה/תנאים כמו left_node_contribs_list, אבל רק שהערך הוא עבור הצומת הימני. |
| רשימה< פלט <מספר שלם>> | רשימת סף () רשימת פלט של טנסור דרגה 1 המציינת את מזהה הדלי שיש להשוות איתו (כסף) לפיצול בכל צומת. |
שיטות בירושה
שיטות ציבוריות
public static BoostedTreesCalculateBestGainsPerFeature create ( scope scope, Operand <Integer> nodeIdRange, Iterable< Operand <Float> > statsSummaryList, Operand <Float> l1, Operand <Float> l2, Operand <Float> treeComplexity>, minNodeSpelights)
שיטת מפעל ליצירת מחלקה העוטפת פעולת BoostedTreesCalculateBestGainsPerFeature חדשה.
פרמטרים
| תְחוּם | ההיקף הנוכחי |
|---|---|
| nodeIdRange | טנסור דירוג 1 (צורה=[2]) כדי לציין את הטווח [ראשון, אחרון) של מזהי צומת לעיבוד בתוך `stats_summary_list`. הצמתים עוברים איטרציה בין שני הצמתים שצוינו על ידי הטנזור, כמו `for node_id in range(node_id_range[0], node_id_range[1])` (שים לב שהאינדקס האחרון node_id_range[1] הוא בלעדי). |
| סטטיסטיקותSummaryList | רשימה של טנזור דירוג 3 (#shape=[max_splits, bucket, 2]) עבור סיכום נתונים סטטיסטיים מצטברים (שיפוע/הסיאן) לכל צומת לכל דליים עבור כל תכונה. המימד הראשון של הטנזור הוא המספר המרבי של פיצולים, ולכן לא כל האלמנטים שלו ישמשו, אלא רק האינדקסים שצוינו על ידי node_ids ישמשו. |
| l1 | גורם רגוליזציה l1 על משקלי עלים, לפי מופע. |
| l2 | גורם רגוליזציה l2 על משקלי עלים, לפי מופע. |
| מורכבות עץ | התאמה לרווח, לפי עלה. |
| minNodeWeight | הממוצע המינימלי של הסינים בצומת נדרש לפני שהצומת ייחשב לפיצול. |
| maxSplits | מספר הצמתים שניתן לפצל בכל העץ. משמש כממד של טנסור פלט. |
החזרות
- מופע חדש של BoostedTreesCalculateBestGainsPerFeature
רשימה ציבורית< פלט <Float>> gainsList ()
רשימת פלט של טנסור דרגה 1 המציינת את הרווחים הטובים ביותר עבור כל תכונה לפיצול עבור צמתים מסוימים. ראה למעלה לפרטים כמו צורות וגדלים.
רשימה ציבורית< פלט <צף>> leftNodeContribsList ()
רשימה של טנסור דרגה 2 המציינת את תרומתם של הצמתים השמאליים בעת הסתעפות מצמתי אב (ניתן על ידי אלמנט הטנזור ב-output node_ids_list) לכיוון השמאלי לפי הסף הנתון עבור כל תכונה. ערך זה ישמש ליצירת ערך הצומת השמאלי על ידי הוספה לערך הצומת האב. גודל הממד השני הוא 1 עבור לוגיטים חד-ממדיים, אך יהיה גדול יותר עבור בעיות מרובות מחלקות. ראה למעלה לפרטים כמו צורות וגדלים.
רשימה ציבורית< פלט <Integer>> nodeIdsList ()
רשימת פלט של טנסור דרגה 1 המציינת מזהי צומת מפוצלים אפשריים עבור כל תכונה. אורך הרשימה הוא num_features, אך לכל טנזור יש גודל שונה מכיוון שכל תכונה מספקת צמתים אפשריים שונים. ראה למעלה לפרטים כמו צורות וגדלים.
רשימה ציבורית< פלט <Float>> rightNodeContribsList ()
רשימה של טנסור דרגה 2, עם אותם צורה/תנאים כמו left_node_contribs_list, אבל רק שהערך הוא עבור הצומת הימני.
רשימה ציבורית< Output <Integer>> thresholdsList ()
רשימת פלט של טנסור דרגה 1 המציינת את מזהה הדלי שיש להשוות איתו (כסף) לפיצול בכל צומת. ראה למעלה לפרטים כמו צורות וגדלים.