s464961/WMICraft
1
0
Fork 0
forked from s464965/WMICraft
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request 'decision_tree_impl' (#39) from decision_tree_impl into master

Browse Source 
Reviewed-on: s464965/WMICraft#39
This commit is contained in:
Jerzy Tomaszewski 2022-06-09 22:33:23 +02:00
commit 08ea0e945e
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)