UniHive_RecSys / archive /README.md
Tirthanker's picture
README: engine v2.0 is the model to use; older models moved to archive/
34bb407 verified
|
Raw History Blame Contribute Delete
1.31 kB
# Archive: earlier models (reference only)
**Do not use these files with `hybrid_engine/`.** The current model is v2.0 in `hybrid_engine/artifacts/` (see the main README). The models here were trained on different inputs (they use community name and flair, user/post IDs or feed position) and have their own vocabularies and standardisation. They are kept so the results in `REPORT.md` can be checked and compared.
| Folder | What it is | Status |
|---|---|---|
| `engine_v1_1/` | the previous hybrid engine (same design, but its models used community name + flair). ONNX, `meta.json`, reports, `pytorch_checkpoints/` (5 seeds per model) | replaced by v2.0 |
| `phase1_models/` | the first five models (LR, DeepFM, Wide&Deep, DCN V2, SASRec), trained with user/post IDs and feed position. ONNX, checkpoints, metadata, `Model_Explainer.docx`, `example_inference.py` | research baseline |
| `v1_two_stage_engine/` | Stage-2 models of the first two-stage engine (tag match, then model). ONNX, checkpoints, Stage-1 weights and results | research baseline, outperformed |
The engine v1.1 files can be loaded on their own with `HybridRecommender(artifacts_dir="archive/engine_v1_1")`, but that version needs content vectors built with its own text recipe ("title. community. flair. body"), not the v2.0 one.