bab992b682
* added dependency on pytest-flask * Updated logging method names * cleaned up S3 configuration for all recommendation engines All S3 configuration data is now in taar.recommenders.s3config * dropped pinned hashes for packages in requirements.txt |
||
---|---|---|
analysis | ||
bin | ||
taar | ||
tests | ||
.flake8 | ||
.gitignore | ||
.travis.yml | ||
DATA_COLLECTION_POLICY.md | ||
LICENSE | ||
MANIFEST.in | ||
Makefile | ||
README.md | ||
environment_emr.yaml | ||
prod-requirements.txt | ||
requirements.txt | ||
setup.cfg | ||
setup.py |
README.md
Taar
Telemetry-Aware Addon Recommender
Table of Contents (ToC):
How does it work?
The recommendation strategy is implemented through the RecommendationManager. Once a recommendation is requested for a specific client id, the recommender iterates through all the registered models (e.g. CollaborativeRecommender) linearly in their registered order. Results are returned from the first module that can perform a recommendation.
Each module specifies its own sets of rules and requirements and thus can decide if it can perform a recommendation independently from the other modules.
Supported models
This is the ordered list of the currently supported models:
Order | Model | Description | Conditions | Generator job |
---|---|---|---|---|
1 | Legacy | recommends WebExtensions based on the reported and disabled legacy add-ons | Telemetry data is available for the user and the user has at least one disabled add-on | source |
2 | Collaborative | recommends add-ons based on add-ons installed by other users (i.e. collaborative filtering) | Telemetry data is available for the user and the user has at least one enabled add-on | source |
3 | Similarity * | recommends add-ons based on add-ons installed by similar representative users | Telemetry data is available for the user and a suitable representative donor can be found | source |
4 | Locale | recommends add-ons based on the top addons for the user's locale | Telemetry data is available for the user and the locale has enough users | source |
5 | Ensemble * | recommends add-ons based on the combined (by stacked generalization) recomendations of other available recommender modules. | More than one of the other Models are available to provide recommendations. | source |
* In order to ensure stable/repeatable testing and prevent unecessary computation, these jobs are not scheduled on Airflow, rather run manually when fresh models are desired.
Instructions for releasing updates
New releases can be shipped by using the normal github workflow. Once a new release is created, it will be automatically uploaded to pypi
.
A note on cdist optimization.
cdist can speed up distance computation by a factor of 10 for the computations we're doing. We can use it without problems on the canberra distance calculation.
Unfortunately there are multiple problems with it accepting a string array. There are different problems in 0.18.1 (which is what is available on EMR), and on later versions. In both cases cdist attempts to convert a string to a double, which fails. For versions of scipy later than 0.18.1 this could be worked around with:
distance.cdist(v1, v2, lambda x, y: distance.hamming(x, y))
However, when you manually provide a callable to cdist, cdist can not do it's baked in
optimizations (https://github.com/scipy/scipy/blob/v1.0.0/scipy/spatial/distance.py#L2408)
so we can just apply the function distance.hamming
to our array manually and get the same
performance.
Build and run tests
You should be able to build taar using Python 2.7 or Python 3.5. To run the testsuite, execute ::
$ python setup.py develop
$ python setup.py test
Alternately, if you've got GNUMake installed, you can just run make test
which will do all of that for you and run flake8 on the codebase.
There are additional integration tests and a microbenchmark available
in tests/test_integration.py
. See the source code for more
information.
Pinning dependencies
TAAR uses hashin (https://pypi.org/project/hashin/) to pin SHA256
hashes for each dependency. To update the hashes, you will need to
remove the run make freeze
which forces all packages in the current
virtualenv to be written out to requirement.txt with versions and SHA
hashes.
Required S3 dependencies
RecommendationManager:
- s3://telemetry-parquet/telemetry-ml/addon_recommender/top_200_whitelist.json
Hybrid Recommender:
- s3://telemetry-parquet/taar/ensemble/ensemble_weight.json
- s3://telemetry-parquet/telemetry-ml/addon_recommender/top_200_whitelist.json
Similarity Recommender:
- s3://telemetry-parquet/taar/similarity/donors.json
- s3://telemetry-parquet/taar/similarity/lr_curves.json
CollaborativeRecommender:
- s3://telemetry-public-analysis-2/telemetry-ml/addon_recommender/item_matrix.json
- s3://telemetry-public-analysis-2/telemetry-ml/addon_recommender/addon_mapping.json
LocaleRecommender:
- s3://telemetry-parquet/taar/locale/top10_dict.json
EnsembleRecommender:
- s3://telemetry-parquet/taar/ensemble/ensemble_weight.json
TAAR breaks out all S3 data load configuration into enviroment
variables. This ensures that running under test has no chance of
clobbering the production data in the event that a developer has AWS
configuration keys installed locally in ~/.aws/
Production enviroment variables required for TAAR
Collaborative Recommender ::
TAAR_ITEM_MATRIX_BUCKET = "telemetry-public-analysis-2"
TAAR_ITEM_MATRIX_KEY = "telemetry-ml/addon_recommender/item_matrix.json"
TAAR_ADDON_MAPPING_BUCKET = "telemetry-public-analysis-2"
TAAR_ADDON_MAPPING_KEY = "telemetry-ml/addon_recommender/addon_mapping.json"
Ensemble Recommender ::
TAAR_ENSEMBLE_BUCKET = "telemetry-parquet"
TAAR_ENSEMBLE_KEY = "taar/ensemble/ensemble_weight.json"
Hybrid Recommender ::
TAAR_WHITELIST_BUCKET = "telemetry-parquet"
TAAR_WHITELIST_KEY = "telemetry-ml/addon_recommender/only_guids_top_200.json"
Locale Recommender ::
TAAR_LOCALE_BUCKET = "telemetry-parquet"
TAAR_LOCALE_KEY = "taar/locale/top10_dict.json"
Similarity Recommender ::
TAAR_SIMILARITY_BUCKET = "telemetry-parquet"
TAAR_SIMILARITY_DONOR_KEY = "taar/similarity/donors.json"
TAAR_SIMILARITY_LRCURVES_KEY = "taar/similarity/lr_curves.json"