Skip to content

Commit

Permalink
refactor: rename output feature predictions() to logits() - 1st step
Browse files Browse the repository at this point in the history
          for generating predictions
  • Loading branch information
jimthompson5802 committed Mar 24, 2020
1 parent b1f25e3 commit e116727
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion ludwig/features/base_feature.py
Original file line number Diff line number Diff line change
Expand Up @@ -167,7 +167,7 @@ def call(
)

# ================ Predictions ================
predictions = self.predictions(feature_hidden)
predictions = self.logits(feature_hidden)

return predictions

Expand Down
2 changes: 1 addition & 1 deletion ludwig/features/binary_feature.py
Original file line number Diff line number Diff line change
Expand Up @@ -154,7 +154,7 @@ def __init__(self, feature):
self._setup_loss()
self._setup_metrics()

def predictions(
def logits(
self,
inputs # hidden
):
Expand Down
2 changes: 1 addition & 1 deletion ludwig/features/numerical_feature.py
Original file line number Diff line number Diff line change
Expand Up @@ -164,7 +164,7 @@ def __init__(self, feature):
self._setup_loss()
self._setup_metrics()

def predictions(
def logits(
self,
inputs, # hidden
):
Expand Down

0 comments on commit e116727

Please sign in to comment.