mirror of
https://github.com/twitter/the-algorithm.git
synced 2025-01-07 01:48:16 +01:00
Compare commits
2 Commits
b4f0518ee5
...
6bbef46224
Author | SHA1 | Date | |
---|---|---|---|
|
6bbef46224 | ||
|
d2da7e56ab |
@ -51,12 +51,13 @@ class TFModelInitializerBuilder:
|
||||
if len(discretized_features) == 0:
|
||||
return
|
||||
|
||||
num_bins = max([len(bins) for bins in discretized_features.values()])
|
||||
num_bins = max(len(bins) for bins in discretized_features.values())
|
||||
|
||||
bin_boundaries_and_weights = {}
|
||||
for feature_name in discretized_features:
|
||||
bin_boundaries_and_weights[feature_name] = self._extract_bin_boundaries_and_weights(
|
||||
bin_boundaries_and_weights = {
|
||||
feature_name: self._extract_bin_boundaries_and_weights(
|
||||
discretized_features[feature_name], num_bins)
|
||||
for feature_name in discretized_features
|
||||
}
|
||||
|
||||
tf_model_initializer["features"]["discretized"] = bin_boundaries_and_weights
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user