s464965/WMICraft
5
2
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

decision_tree_impl #39

Merged
s464981 merged 2 commits from decision_tree_impl into master 2022-06-09 22:33:24 +02:00
Conversation 0 Commits 2 Files Changed 3 +1003 -2
3 changed files with 1003 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1001
learning/dataset_tree_1000.csv Normal file
View File

File diff suppressed because it is too large Load Diff

0
learning/dataset_tree.csv → learning/dataset_tree_300.csv
View File

2
learning/decision_tree.py
View File

@ -26,7 +26,7 @@ def parse_idx_of_opp_or_monster(s: str) -> int:
class DecisionTree: class DecisionTree:
def __init__(self) -> None: def __init__(self) -> None:
data_frame = pd.read_csv('learning/dataset_tree.csv', delimiter=';') data_frame = pd.read_csv('learning/dataset_tree_1000.csv', delimiter=';')
unlabeled_goals = data_frame['goal'] unlabeled_goals = data_frame['goal']
self.goals_label_encoder = LabelEncoder() self.goals_label_encoder = LabelEncoder()
self.goals = self.goals_label_encoder.fit_transform(unlabeled_goals) self.goals = self.goals_label_encoder.fit_transform(unlabeled_goals)