issues
search
zerebom
/
H-and-M-Personalized-Fashion-Recommendations
(133rd/2952🥈)
4
stars
2
forks
source link
Add cache & emb_mean_vec block
#18
Closed
zerebom
closed
2 years ago
zerebom
commented
2 years ago
Why
高速に実験をしたい
embベクトルとユーザのemb平均ベクトルと類似度を取るクラスを作りたい
Why
キャッシュ機構の追加
embの平均ベクトルと類似度を取るクラスを作りたい
↑これらを盛り込んだ実験ファイル016...の追加
キャッシュ機構の説明
概要
Abstract methodに書かれているdirにpkl形式で保存する
use_cacheフラグがTrue & すでにファイルが保存されているときはファイルを読み出す
それ以外の時は普通にtrasformを実行してファイル保存する
基本的にクラス名+target_weekでファイル名を保存するが、TargetEncodingとか同じクラスを複数呼ぶ場合はindex_colなどで呼びわける
使い方
普通にクラス呼び出し時にuse_chaceを渡すだけでOK
これまでの引数にtarget_weekを追加。target_weekによらないCG, FEはNoneなどを文字列で渡せばOK
CG, FEの実装時も他のクラスを真似して書けばOK
子クラスは基本的に引数の変更だけでOK
Dis2HistoryAveVecAndArtVecの説明
概要
ユーザのインタラクションに対してアーティクルのベクトルを紐付け平均をとる → ユーザの平均ベクトルを作成
base_dfに存在するユーザ、アーティクルペアに対してそれぞれのベクトルの類似度をGPUで計算する
使い方
{article_id(str):emb([List[float]])}辞書を渡す
Why
Why
キャッシュ機構の説明
Dis2HistoryAveVecAndArtVecの説明