Scribd Logo
Upload

Welcome to Scribd!

  • 146 views

    Assignment 1 Data Mining

    Uploaded by

    Aldi Renadi
    This document shows code for building a decision tree classifier model to predict loan defaults. It loads and preprocesses a CSV dataset containing home ownership, marital status, income, and default labels. It splits the data into features and a target, fits a decision tree classifier, exports a visualization of the tree, and makes predictions on new data points.

    Copyright:

    © All Rights Reserved
    Download as PDF, TXT or read online from Scribd

    Assignment 1 Data Mining

    Uploaded by

    Aldi Renadi
    146 views1 page
    This document shows code for building a decision tree classifier model to predict loan defaults. It loads and preprocesses a CSV dataset containing home ownership, marital status, income, and default labels. It splits the data into features and a target, fits a decision tree classifier, exports a visualization of the tree, and makes predictions on new data points.

    Copyright

    © © All Rights Reserved

    Available Formats

    PDF, TXT or read online from Scribd

    Did you find this document useful?

    Is this content inappropriate?

    This document shows code for building a decision tree classifier model to predict loan defaults. It loads and preprocesses a CSV dataset containing home ownership, marital status, income, and default labels. It splits the data into features and a target, fits a decision tree classifier, exports a visualization of the tree, and makes predictions on new data points.

    Copyright:

    © All Rights Reserved
    Download as PDF, TXT or read online from Scribd
    Download as pdf or txt
    146 views1 page

    Assignment 1 Data Mining

    Uploaded by

    Aldi Renadi
    This document shows code for building a decision tree classifier model to predict loan defaults. It loads and preprocesses a CSV dataset containing home ownership, marital status, income, and default labels. It splits the data into features and a target, fits a decision tree classifier, exports a visualization of the tree, and makes predictions on new data points.

    Copyright:

    © All Rights Reserved
    Download as PDF, TXT or read online from Scribd
    Download as pdf or txt
    of 1

    In 

    [1]:
    #Import Module

    import pandas as pd

    from sklearn import tree

    from sklearn.tree import DecisionTreeClassifier

    In [2]:
    #Import and Read Dataset

    dataset = pd.read_csv('https://raw.githubusercontent.com/asrulabdullah99/data_mining/master/dataset_decision/Dataset_Tugas.csv')

    dataset

    Out[2]: Home_Owner Marital_Status Annual_Income Defaulted_Borrower

    0 Yes Single 125000 No

    1 No Married 100000 No

    2 No Single 70000 No

    3 Yes Married 120000 No

    4 No Divorced 95000 Yes

    5 No Married 60000 No

    6 Yes Divorced 220000 No

    7 No Single 85000 Yes

    8 No Married 75000 No

    9 No Single 90000 Yes

    In [3]:
    #Convert all data to numerical values

    d = {'Yes':1,'No':0}

    dataset['Home_Owner']=dataset['Home_Owner'].map(d)

    d = {'Single':0,'Married':1,'Divorced':2}

    dataset['Marital_Status']=dataset['Marital_Status'].map(d)

    d = {'Yes':1,'No':0}

    dataset['Defaulted_Borrower']=dataset['Defaulted_Borrower'].map(d)

    dataset

    Out[3]: Home_Owner Marital_Status Annual_Income Defaulted_Borrower

    0 1 0 125000 0

    1 0 1 100000 0

    2 0 0 70000 0

    3 1 1 120000 0

    4 0 2 95000 1

    5 0 1 60000 0

    6 1 2 220000 0

    7 0 0 85000 1

    8 0 1 75000 0

    9 0 0 90000 1

    In [4]:
    #Saparate columns

    #x adalah feature columns, y = target columns

    features = ['Home_Owner','Marital_Status','Annual_Income','Defaulted_Borrower']

    x = dataset[features]

    y = dataset['Defaulted_Borrower']

    print(x)

    print(y)

    Home_Owner Marital_Status Annual_Income Defaulted_Borrower

    0 1 0 125000 0

    1 0 1 100000 0

    2 0 0 70000 0

    3 1 1 120000 0

    4 0 2 95000 1

    5 0 1 60000 0

    6 1 2 220000 0

    7 0 0 85000 1

    8 0 1 75000 0

    9 0 0 90000 1

    0 0

    1 0

    2 0

    3 0

    4 1

    5 0

    6 0

    7 1

    8 0

    9 1

    Name: Defaulted_Borrower, dtype: int64

    In [5]:
    import pydotplus

    import matplotlib.pyplot as plt

    import matplotlib.image as pltimg

    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('decisiontree.png')

    img = pltimg.imread('decisiontree.png')

    imgplot = plt.imshow(img)

    plt.show()

    In [6]:
    #Prediksi dari umur 39, 10 tahun experience, comedy ranking 7 dan USA

    print(dtree.predict([[39,15,3,0]]))

    [0]

    In [8]:
    print(dtree.predict([[1, 0, 125000, 0]]))

    [0]

    In [ ]:

    You might also like