forked from s464965/WMICraft
Merge pull request 'decision_tree_impl' (#39) from decision_tree_impl into master
Reviewed-on: s464965/WMICraft#39
This commit is contained in:
commit
08ea0e945e
1001
learning/dataset_tree_1000.csv
Normal file
1001
learning/dataset_tree_1000.csv
Normal file
File diff suppressed because it is too large
Load Diff
@ -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)
|
||||||
|
Loading…
Reference in New Issue
Block a user