s481808/Traktor
4
2
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Decision_trees #3

Merged
s481838 merged 3 commits from Decision_trees into master 2024-05-20 23:11:52 +02:00
Conversation 0 Commits 3 Files Changed 9 +1 -3
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 4693c3e980 - Show all commits

2
source/decision_tree.py
View File

@ -23,8 +23,6 @@ X = data.drop('target_column', axis=1)
y = data['target_column'] y = data['target_column']
X = pd.get_dummies(X) X = pd.get_dummies(X)
print(X)
print(y)
# Podział danych na zbiór treningowy i testowy # Podział danych na zbiór treningowy i testowy
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42) X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)