Skip to content
Snippets Groups Projects
Commit aa451672 authored by chsieh16's avatar chsieh16
Browse files

Annotate the memorized coefficient maps

parent 3bcef547
No related branches found
No related tags found
No related merge requests found
......@@ -3,7 +3,7 @@ import csv
import itertools
import json
import os
from typing import Any, List, Optional, Tuple
from typing import Any, Dict, List, Optional, Tuple
import numpy as np
......@@ -19,7 +19,10 @@ class DTreeLearner(LearnerBase):
self._perc_dim: int = perc_dim
self._s2f_func = lambda x: x
self._var_coeff_map = {}
# Given a (derived) feature name,
# this map returns a mapping from base feature names to coefficients
self._var_coeff_map: Dict[str, Dict[str, int]] = {}
# check directory name exists, if not create it.
self.dir_name = "out"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment