Question

我在大学里有一个制定决策树的项目，我已经有创建决策树的代码，但是我想打印它，有人可以帮助我吗？

#IMPORT ALL NECESSARY LIBRARIES
import Chefboost as chef
import pandas as pd

archivo = input("INSERT FILE NAMED FOLLOWED BY .CSV:\n")
# READ THE DATA SET FROM THE CSV FILE
df = pd.read_csv(str(archivo))
df.columns = ['ph', 'soil_temperature', 'soil_moisture', 'illuminance', 'env_temperature','env_humidity','Decision']
# print(df.head(10)) #UNCOMMENT IF WANT FIRST 10 ROWS PRINTED OUT


config = {'algorithm':'ID3'} # CONFIGURE THE ALGORITH. CHOOSE BETWEEN ID3, C4.5, CART, Regression
model = chef.fit(df.copy(), config) #CREATE THE DECISION TREE BASED OF THE CONFIGURATION ABOVE

resultados = pd.DataFrame(columns = ["Real", "Predicción"]) #CREATE AN EMPTY PANDAS DATAFRAME
# SAVE ALL REAL VS ESTIMATED VALUES IN THE ABOVE DATAFRAME
for i in range(1,372):
    l = []
    l.append(df.iloc[i]['Decision'])
    feature = df.iloc[i]
    prediction = chef.predict(model, feature)
    l.append(prediction)
    resultados.loc[i] = l
    print(l)

Answer 1

不知道Chefboost库，我无法直接回答您的问题，但是当我使用新库时，我经常会使用一些工具来帮助我了解库给我的东西。使用dir（object）获取对象的属性和方法的列表。

当您“打印决策树”时，您可能还会想知道些什么。您是否要打印模型或预测？您遇到什么麻烦或看到什么错误？

希望这会有所帮助。

使用Python打印决策树

1 个答案: