decision_tree_impl #39

Merged
s464981 merged 2 commits from decision_tree_impl into master 2022-06-09 22:33:24 +02:00
3 changed files with 1003 additions and 2 deletions

File diff suppressed because it is too large Load Diff

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)