Advertisement
yesh666

id3

Mar 25th, 2023 (edited)
769
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.72 KB | None | 0 0
  1. import pandas as pd
  2. from sklearn.tree import DecisionTreeClassifier
  3. from sklearn.model_selection import train_test_split
  4.  
  5. data = pd.read_csv('data.csv')
  6. X = data.iloc[:, :-1]
  7. y = data.iloc[:, -1]
  8. X = pd.get_dummies(X)
  9. X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=1)
  10.  
  11. clf = DecisionTreeClassifier()
  12. clf.fit(X_train, y_train)
  13.  
  14. print("Accuracy:", clf.score(X_test, y_test))
  15.  
  16. new_data = pd.DataFrame({
  17.     'outlook': ['overcast'],
  18.     'temp': ['mild'],
  19.     'humidity': ['high'],
  20.     'windy': ['FALSE']
  21. })
  22. new_data = pd.get_dummies(new_data)
  23. new_data = new_data.reindex(columns=X.columns, fill_value=0)
  24. prediction = clf.predict(new_data)
  25.  
  26. print("Prediction:", prediction[0])
  27.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement