Skip to content

Commit

Permalink
chore[rfctr]: change variable name
Browse files Browse the repository at this point in the history
  • Loading branch information
ME-researchgroup committed Feb 14, 2024
1 parent 80d5456 commit f1bdd96
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions weightfactors/raking/generalized_raker.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@ def __init__(
self.raise_on_extreme = raise_on_extreme
self.cutoffs = cutoffs
self.exclusion_column = exclusion_column
self.question_names = list(population_targets.keys())
self.column_names = list(population_targets.keys())
self.loss: List[float] = [np.inf]
self.success: bool = False
self.weights = np.empty(0)
Expand All @@ -83,7 +83,7 @@ def create_design_matrix(self, data: pd.DataFrame) -> np.ndarray:
... [0 1]
... [0 1]]
"""
dummies = pd.get_dummies(data, columns=self.question_names)
dummies = pd.get_dummies(data, columns=self.column_names)
design_matrix = dummies[dummies.columns.difference(data.columns, sort=False)]
return np.asarray(design_matrix.values)

Expand All @@ -98,7 +98,7 @@ def create_target_vector(self, data: pd.DataFrame) -> np.ndarray:
"""
target_vector = []
for question in self.question_names:
for question in self.column_names:
counts = data[question].value_counts(sort=False) # Don't sort by frequency
total_observed_n = counts.sum()
sorted_counts = dict(sorted(counts.items())) # Sort keys alphabetically
Expand Down

0 comments on commit f1bdd96

Please sign in to comment.