Mirrors/the-algorithm
2
0
Fork 0
mirror of https://github.com/twitter/the-algorithm.git synced 2024-06-02 17:28:45 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge d2da7e56ab into 72eda9a24f

Browse Source
This commit is contained in:
JHV 2023-07-17 21:38:12 -05:00 committed by GitHub
commit b4f0518ee5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 68 additions and 67 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/python/twitter/deepbird/projects/timelines/scripts/models/earlybird/lolly/tf_model_initializer_builder.py
View File

@ -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