Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/SAI990323/BIGRec
/ functions
Functions
22 in github.com/SAI990323/BIGRec
⨍
Functions
22
◇
Types & classes
0
↓ 3 callers
Function
get_ndcg
(test_data, rank)
data/movie/adjust_pda.py:65
↓ 2 callers
Function
batch
(list, batch_size=batch_size)
inference.py:133
↓ 2 callers
Function
generate_and_tokenize_prompt
(data_point)
train.py:148
↓ 2 callers
Function
generate_prompt
(data_point)
train.py:281
↓ 2 callers
Function
get_dist
(test_data)
data/game/adjust_ci.py:40
↓ 2 callers
Function
get_dist
(test_data)
data/movie/adjust_ci.py:38
↓ 2 callers
Function
get_dist
(test_data)
data/movie/adjust_pda.py:38
↓ 2 callers
Function
get_hr
(test_data, rank)
data/game/adjust_ci.py:81
↓ 2 callers
Function
get_hr
(test_data, rank)
data/movie/adjust_ci.py:81
↓ 2 callers
Function
get_ndcg
(test_data, rank)
data/game/adjust_ci.py:67
↓ 2 callers
Function
get_ndcg
(test_data, rank)
data/movie/adjust_ci.py:65
↓ 2 callers
Function
tokenize
(prompt, add_eos_token=True)
train.py:126
↓ 1 callers
Function
batch
(list, batch_size=1)
data/game/adjust_ci.py:45
↓ 1 callers
Function
batch
(list, batch_size=1)
data/game/evaluate.py:57
↓ 1 callers
Function
batch
(list, batch_size=1)
data/movie/adjust_ci.py:43
↓ 1 callers
Function
batch
(list, batch_size=1)
data/movie/adjust_pda.py:43
↓ 1 callers
Function
batch
(list, batch_size=1)
data/movie/evaluate.py:49
↓ 1 callers
Function
evaluate
( instructions, inputs=None, temperature=0, top_p=0.9, top_k=40,
inference.py:92
↓ 1 callers
Function
generate_prompt
(instruction, input=None)
inference.py:149
↓ 1 callers
Function
get_hr
(test_data, rank)
data/movie/adjust_pda.py:81
Function
main
( load_8bit: bool = False, base_model: str = "", lora_weights: str = "tloen/alpaca-lora-7b", t
inference.py:29
Function
train
( # model/data params base_model: str = "", # the only required argument train_data_path: List[st
train.py:29