BoostedTreesTrainingPredict

คลาสสุดท้ายสาธารณะ BoostedTreesTrainingPredict

รันตัวทำนายชุดการถดถอยแบบบวกหลายตัวบนอินสแตนซ์อินพุตและ

คำนวณการอัปเดตเป็นบันทึกที่แคชไว้ ออกแบบมาเพื่อใช้ระหว่างการฝึก โดยจะสำรวจทรีโดยเริ่มจาก id ทรีที่แคชและ id โหนดที่แคช และคำนวณการอัปเดตที่จะพุชไปยังแคช

วิธีการสาธารณะ

BoostedTreesTrainingPredict แบบคงที่
สร้าง (ขอบเขต ขอบเขต ตัวดำเนินการ <?> treeEnsembleHandle, ตัวดำเนินการ <Integer> cachedTreeIds, ตัวดำเนินการ <Integer> cachedNodeIds, Iterable < ตัวดำเนินการ <Integer>> bucketizedFeatures, logitsDimension แบบยาว)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ BoostedTreesTrainingPredict ใหม่
เอาต์พุต <จำนวนเต็ม>
รหัสโหนด ()
อันดับ 1 เทนเซอร์ที่มีรหัสโหนดใหม่ใน tree_ids ใหม่
เอาท์พุต <โฟลต>
บันทึกบางส่วน ()
Tensor อันดับ 2 ที่มีการอัปเดต logits (เกี่ยวกับค่าแคชที่เก็บไว้) สำหรับแต่ละตัวอย่าง
เอาต์พุต <จำนวนเต็ม>
รหัสต้นไม้ ()
เทนเซอร์อันดับ 1 ที่มีรหัสต้นไม้ใหม่สำหรับแต่ละตัวอย่าง

วิธีการสืบทอด

วิธีการสาธารณะ

สร้าง BoostedTreesTrainingPredict แบบคงที่สาธารณะ (ขอบเขต ขอบเขต ตัวดำเนินการ <?> treeEnsembleHandle, ตัวดำเนินการ <Integer> cachedTreeIds, ตัวดำเนินการ <Integer> cachedNodeIds, Iterable < ตัวดำเนินการ <Integer>> bucketizedFeatures, LogitsDimension แบบยาว)

วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ BoostedTreesTrainingPredict ใหม่

พารามิเตอร์
ขอบเขต ขอบเขตปัจจุบัน
cachedTreeIds อันดับ 1 Tensor ที่มีรหัสทรีแคชซึ่งเป็นแผนผังเริ่มต้นของการทำนาย
cachedNodeIds อันดับ 1 Tensor ที่มีรหัสโหนดแคชซึ่งเป็นโหนดเริ่มต้นของการทำนาย
คุณสมบัติที่เก็บไว้ในถัง รายการเทนเซอร์อันดับ 1 ที่มีรหัสบัคเก็ตสำหรับแต่ละฟีเจอร์
บันทึกมิติข้อมูล สเกลาร์ ซึ่งเป็นมิติของบันทึก ที่จะใช้สำหรับรูปร่างบันทึกบางส่วน
การส่งคืน
  • อินสแตนซ์ใหม่ของ BoostedTreesTrainingPredict

สาธารณะ ออก <จำนวนเต็ม> nodeIds ()

อันดับ 1 เทนเซอร์ที่มีรหัสโหนดใหม่ใน tree_ids ใหม่

เอาท์พุท สาธารณะ <Float> บางส่วน Logits ()

Tensor อันดับ 2 ที่มีการอัปเดต logits (เกี่ยวกับค่าแคชที่เก็บไว้) สำหรับแต่ละตัวอย่าง

สาธารณะ เอาท์พุท <จำนวนเต็ม> treeIds ()

เทนเซอร์อันดับ 1 ที่มีรหัสต้นไม้ใหม่สำหรับแต่ละตัวอย่าง