西瓜决策树python代码

西瓜书中提到了C4.5算法和CART算法，它们都是决策树的经典算法。如果你想了解决策树算法的python实现，可以使用scikit-learn库中的DecisionTreeClassifier类。下面是一个简单的例子：

from sklearn.tree import DecisionTreeClassifier
from sklearn.datasets import load_iris

# 加载数据集
iris = load_iris()

# 创建决策树分类器对象
clf = DecisionTreeClassifier()

# 训练模型
clf.fit(iris.data, iris.target)

# 预测新样本
print(clf.predict([[5.1, 3.5, 1.4, 0.2]]))

以上代码使用iris数据集进行训练，然后预测新样本。你可以根据需要修改数据集和决策树分类器的参数。

相关问题

西瓜分类决策树python

决策树是一种常用的机器学习算法，用于分类和回归问题。它通过构建一棵树状结构来进行决策，每个内部节点表示一个特征或属性，每个叶子节点表示一个类别或值。

在Python中，可以使用scikit-learn库来实现决策树算法。下面是一个简单的示例代码，用于构建和训练一个西瓜分类决策树模型：

# 导入所需的库
from sklearn import datasets
from sklearn.tree import DecisionTreeClassifier
from sklearn.model_selection import train_test_split

# 加载西瓜数据集
data = datasets.load_wine()
X = data.data  # 特征
y = data.target  # 标签

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

# 构建决策树模型
model = DecisionTreeClassifier()

# 训练模型
model.fit(X_train, y_train)

# 在测试集上进行预测
y_pred = model.predict(X_test)

# 输出预测结果
print("预测结果：", y_pred)

# 输出模型准确率
accuracy = model.score(X_test, y_test)
print("模型准确率：", accuracy)

这段代码使用了sklearn库中的`DecisionTreeClassifier`类来构建决策树模型，并使用`train_test_split`函数将数据集划分为训练集和测试集。然后，通过调用`fit`方法对模型进行训练，再使用`predict`方法对测试集进行预测。最后，通过`score`方法计算模型在测试集上的准确率。

西瓜数据集决策树python不调库代码实现

以下是使用Python实现西瓜数据集决策树的代码，不需要调用任何库：

import math

# 定义节点类
class Node:
    def __init__(self, feature=None, value=None, results=None, tb=None, fb=None):
        self.feature = feature  # 用于划分数据集的特征
        self.value = value  # 特征的值
        self.results = results  # 存储叶子节点的分类结果
        self.tb = tb  # 左子树
        self.fb = fb  # 右子树

# 计算数据集的熵
def entropy(data):
    results = {}
    for row in data:
        r = row[-1]
        if r not in results:
            results[r] = 0
        results[r] += 1
    ent = 0.0
    for r in results:
        p = float(results[r]) / len(data)
        ent -= p * math.log(p, 2)
    return ent

# 根据特征和特征值划分数据集
def divide_data(data, feature, value):
    split_func = None
    if isinstance(value, int) or isinstance(value, float):
        split_func = lambda row: row[feature] >= value
    else:
        split_func = lambda row: row[feature] == value
    set1 = [row for row in data if split_func(row)]
    set2 = [row for row in data if not split_func(row)]
    return (set1, set2)

# 选择最好的特征和特征值来划分数据集
def find_best_feature(data):
    best_feature = -1
    best_value = None
    best_gain = 0.0
    base_entropy = entropy(data)
    for feature in range(len(data[0]) - 1):
        feature_values = set([row[feature] for row in data])
        for value in feature_values:
            set1, set2 = divide_data(data, feature, value)
            p = float(len(set1)) / len(data)
            gain = base_entropy - p * entropy(set1) - (1 - p) * entropy(set2)
            if gain > best_gain:
                best_feature = feature
                best_value = value
                best_gain = gain
    return (best_feature, best_value)

# 构建决策树
def build_tree(data):
    if len(data) == 0:
        return Node()
    results = [row[-1] for row in data]
    if results.count(results[0]) == len(results):
        return Node(results=results[0])
    best_feature, best_value = find_best_feature(data)
    set1, set2 = divide_data(data, best_feature, best_value)
    tb = build_tree(set1)
    fb = build_tree(set2)
    return Node(feature=best_feature, value=best_value, tb=tb, fb=fb)

# 打印决策树
def print_tree(tree, indent=''):
    if tree.results is not None:
        print(str(tree.results))
    else:
        print(str(tree.feature) + ':' + str(tree.value) + '? ')
        print(indent + 'T->', end='')
        print_tree(tree.tb, indent + '  ')
        print(indent + 'F->', end='')
        print_tree(tree.fb, indent + '  ')

# 对新数据进行分类
def classify(tree, data):
    if tree.results is not None:
        return tree.results
    else:
        v = data[tree.feature]
        branch = None
        if isinstance(v, int) or isinstance(v, float):
            if v >= tree.value:
                branch = tree.tb
            else:
                branch = tree.fb
        else:
            if v == tree.value:
                branch = tree.tb
            else:
                branch = tree.fb
        return classify(branch, data)

# 测试决策树
def test_tree(tree, test_data):
    correct = 0
    for row in test_data:
        if classify(tree, row[:-1]) == row[-1]:
            correct += 1
    accuracy = float(correct) / len(test_data)
    print('Accuracy: %.2f%%' % (accuracy * 100))

# 加载西瓜数据集
def load_watermelon():
    data = [
        [1, 1, 1, 1, 'yes'],
        [1, 1, 1, 0, 'yes'],
        [1, 0, 1, 0, 'no'],
        [0, 1, 0, 1, 'no'],
        [0, 1, 0, 0, 'no'],
        [0, 0, 1, 1, 'no'],
        [0, 1, 1, 0, 'no'],
        [1, 1, 0, 1, 'no'],
        [1, 0, 0, 0, 'no'],
        [0, 1, 0, 1, 'no']
    ]
    return data

# 加载西瓜数据集2
def load_watermelon2():
    data = [
        [0.697, 0.460, 1, 'yes'],
        [0.774, 0.376, 1, 'yes'],
        [0.634, 0.264, 1, 'yes'],
        [0.608, 0.318, 1, 'yes'],
        [0.556, 0.215, 1, 'yes'],
        [0.403, 0.237, 1, 'yes'],
        [0.481, 0.149, 1, 'yes'],
        [0.437, 0.211, 1, 'yes'],
        [0.666, 0.091, 0, 'no'],
        [0.243, 0.267, 0, 'no'],
        [0.245, 0.057, 0, 'no'],
        [0.343, 0.099, 0, 'no'],
        [0.639, 0.161, 0, 'no'],
        [0.657, 0.198, 0, 'no'],
        [0.360, 0.370, 0, 'no'],
        [0.593, 0.042, 0, 'no'],
        [0.719, 0.103, 0, 'no']
    ]
    return data

# 加载西瓜数据集3
def load_watermelon3():
    data = [
        ['青绿', '蜷缩', '浊响', '清晰', '凹陷', '硬滑', '是'],
        ['乌黑', '蜷缩', '沉闷', '清晰', '凹陷', '硬滑', '是'],
        ['乌黑', '蜷缩', '浊响', '清晰', '凹陷', '硬滑', '是'],
        ['青绿', '稍蜷', '浊响', '清晰', '稍凹', '软粘', '是'],
        ['浅白', '稍蜷', '浊响', '清晰', '稍凹', '软粘', '是'],
        ['青绿', '硬挺', '清脆', '清晰', '平坦', '软粘', '是'],
        ['乌黑', '稍蜷', '沉闷', '稍糊', '稍凹', '硬滑', '是'],
        ['乌黑', '稍蜷', '浊响', '清晰', '稍凹', '软粘', '是'],
        ['乌黑', '稍蜷', '沉闷', '稍糊', '稍凹', '硬滑', '否'],
        ['青绿', '硬挺', '清脆', '清晰', '平坦', '硬滑', '否'],
        ['浅白', '硬挺', '清脆', '模糊', '平坦', '软粘', '否'],
        ['浅白', '蜷缩', '浊响', '模糊', '平坦', '硬滑', '否'],
        ['青绿', '稍蜷', '浊响', '稍糊', '凹陷', '硬滑', '否'],
        ['浅白', '稍蜷', '沉闷', '稍糊', '凹陷', '硬滑', '否'],
        ['乌黑', '稍蜷', '浊响', '清晰', '稍凹', '软粘', '否'],
        ['浅白', '蜷缩', '浊响', '模糊', '平坦', '硬滑', '否'],
        ['青绿', '蜷缩', '沉闷', '稍糊', '稍凹', '硬滑', '否']
    ]
    return data

# 加载西瓜数据集4
def load_watermelon4():
    data = [
        ['青绿', '蜷缩', '浊响', '清晰', '凹陷', '硬滑', '是'],
        ['乌黑', '蜷缩', '沉闷', '清晰', '凹陷', '硬滑', '是'],
        ['乌黑', '蜷缩', '浊响', '清晰', '凹陷', '硬滑', '是'],
        ['青绿', '稍蜷', '浊响', '清晰', '稍凹', '软粘', '是'],
        ['浅白', '稍蜷', '浊响', '清晰',