Decision Tree Visualizer
v1.0.1
Tree Visualizer เป็นห้องสมุดที่ทรงพลังที่ช่วยให้คุณเห็นภาพตัวจําแนกต้นไม้ตัดสินใจ Sklearn ได้อย่างง่ายดาย มันมีฟังก์ชั่นสำหรับการแยกข้อมูลที่เป็นประโยชน์เกี่ยวกับโครงสร้างต้นไม้และกฎและสร้างไฟล์ HTML สำหรับการแสดงภาพต้นไม้การตัดสินใจ
ในการติดตั้งไลบรารีให้ใช้ PIP:
pip install d-treevis
ในการเริ่มต้นใช้งานนำเข้าไลบรารีและใช้ฟังก์ชั่น create_tree และ create_sankey :
import d_treevis as dtv ถัดไปพอดีกับตัวจําแนกทรีตัดสินใจของ Sklearn บนชุดข้อมูลของคุณและส่งผ่านไปยังฟังก์ชั่น create_tree :
from sklearn . tree import DecisionTreeClassifier
# Fit a Decision Tree Classifier on your dataset
tree_model = DecisionTreeClassifier ()
tree_model . fit ( X , y )
# Visualize the decision tree
visualizer . visualize ( tree_model )นอกจากนี้คุณยังสามารถปรับแต่งชื่อและสีเป้าหมายเพื่อการสร้างภาพที่ดีขึ้น:
# Define target names and colors
target_names = [ 'Survived' , 'Not Survived' ]
target_colors = [ 'red' , 'yellow' ]
# Visualize the decision tree with custom target names and colors
visualizer . visualize ( tree_model , target_names = target_names , target_colors = target_colors )สำหรับเอกสารและตัวอย่างโดยละเอียดโปรดดูเอกสารการตัดสินใจของทรีทรีตัดสินใจ
ห้องสมุดนี้ได้รับใบอนุญาตภายใต้ใบอนุญาต MIT
ยินดีต้อนรับ! โปรดอ่านแนวทางการบริจาคก่อนส่งคำขอดึง
หากคุณมีคำถามหรือข้อเสนอแนะอย่าลังเลที่จะติดต่อเราที่ [email protected]
