Jump to content

andretkj1

Members
  • Posts

    1
  • Joined

  • Last visited

Posts posted by andretkj1

  1. try ing this :
    https://www.w3schools.com/python/showpython.asp?filename=demo_ml_dtree4

    import pandas
    from sklearn import tree
    import pydotplus
    from sklearn.tree import DecisionTreeClassifier
    import matplotlib.pyplot as plt
    import matplotlib.image as pltimg
    
    df = pandas.read_csv("shows.csv")
    
    d = {'UK': 0, 'USA': 1, 'N': 2}
    df['Nationality'] = df['Nationality'].map(d)
    d = {'YES': 1, 'NO': 0}
    df['Go'] = df['Go'].map(d)
    
    features = ['Age', 'Experience', 'Rank', 'Nationality']
    
    X = df[features]
    y = df['Go']
    
    dtree = DecisionTreeClassifier()
    dtree = dtree.fit(X, y)
    data = tree.export_graphviz(dtree, out_file=None, feature_names=features)
    graph = pydotplus.graph_from_dot_data(data)
    graph.write_png('mydecisiontree.png') # error in this line, got the error like in the image

    trying this but got this error

    image.png.bedefe7adfcda064f60dff8e1511f9a7.png

×
×
  • Create New...