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

Function TFModelInitializerBuilder._set_discretized_features refactored with the following changes:

Browse Source 
removes a pair of brackets, making the intent slightly clearer
 Convert for loop into dictionary comprehension (dict-comprehension)
This commit is contained in:
Jhv 2023-04-04 21:28:36 +08:00
parent d1cab28a10
commit d2da7e56ab
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