Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/HKUDS/AI-Researcher
/ functions
Functions
951 in github.com/HKUDS/AI-Researcher
⨍
Functions
951
◇
Types & classes
147
↳
Endpoints
10
↓ 3 callers
Method
neg_sampling
(self)
examples/dccf/project/data_processing/dataset.py:13
↓ 3 callers
Function
plot_training_curves
Plot training and validation curves.
examples/fsq/project/visualize_results.py:15
↓ 3 callers
Method
pretty_print_messages
(self, message, **kwargs)
research_agent/inno/logger.py:92
↓ 3 callers
Method
read_section_content
Read content from an existing section file
paper_agent/introduction_composing.py:15
↓ 3 callers
Method
read_section_content
Read content from an existing section file
paper_agent/conclusion_composing.py:15
↓ 3 callers
Method
read_section_content
Read content from an existing section file
paper_agent/abstract_composing.py:16
↓ 3 callers
Method
register
统一的注册装饰器 Args: type: 注册类型,"tool" 或 "agent" name: 可选的注册名称
research_agent/inno/registry.py:40
↓ 3 callers
Method
run_async
( self, agent: Agent, messages: List, context_variables: dict = {}, mo
research_agent/inno/core.py:481
↓ 3 callers
Function
run_command_in_container
communicate with docker container and execute command, support stream output Args: command: the command to execute strea
research_agent/inno/util.py:319
↓ 3 callers
Function
run_experiment
(config)
examples/hgcl/project/run_experiments.py:37
↓ 3 callers
Method
sample
Generate samples using Euler method
examples/con_flowmatching/project/model/network.py:49
↓ 3 callers
Function
should_continue
()
research_agent/inno/environment/shutdown_listener.py:45
↓ 3 callers
Function
should_exit
()
research_agent/inno/environment/shutdown_listener.py:40
↓ 3 callers
Method
velocity_field
Compute velocity field v_θ(t, x_t)
examples/con_flowmatching/project/model/network.py:45
↓ 3 callers
Function
visualize_results
(data_dict, xlabel, ylabel, title, save_path)
examples/hgcl/project/run_experiments.py:21
↓ 3 callers
Function
with_env
将env注入到工具函数中的装饰器
research_agent/inno/environment/docker_env.py:195
↓ 2 callers
Method
__init__
(self, input_dim, num_heads=4)
examples/gnn_nodeformer/project/model/enhanced_model.py:11
↓ 2 callers
Method
__init__
Encoder network for VAE. Args: in_channels (int): Number of input channels latent_dim (int): Dimension of lat
examples/fsq/project/model/vae.py:7
↓ 2 callers
Method
__init__
(self, dim, activation='relu')
examples/con_flowmatching/project/model/resnet_velocity.py:7
↓ 2 callers
Function
_checkMeetChallenge
check if meet challenge Examples: _checkMeetChallenge()
research_agent/inno/environment/browser_env.py:275
↓ 2 callers
Method
_convert_docker_to_local
(self, path: str)
research_agent/inno/environment/markdown_browser/requests_markdown_browser.py:137
↓ 2 callers
Method
_extract_text_with_docling
Extract text using Docling backend.
benchmark_collection/utils/pdf_utils.py:168
↓ 2 callers
Method
_find_next_viewport
Search for matches between the starting viewport looping when reaching the end.
research_agent/inno/environment/markdown_browser/requests_markdown_browser.py:267
↓ 2 callers
Function
_fix_stopword
Fix the issue when some LLM would NOT return the stopword.
research_agent/inno/fn_call_converter.py:565
↓ 2 callers
Method
_handle_rate_limit
处理 API 速率限制
research_agent/inno/tools/github_client.py:248
↓ 2 callers
Function
_local_to_docker
Convert a local path to a docker path local_path: the local path to convert, like `{local_workplace}/downloads/xxx` docker_path: the dock
research_agent/inno/environment/browser_env.py:45
↓ 2 callers
Function
_register_signal_handlers
()
research_agent/inno/environment/shutdown_listener.py:28
↓ 2 callers
Method
_transcribe_audio
(self, local_path)
research_agent/inno/environment/mdconvert.py:653
↓ 2 callers
Method
_transcribe_audio
(self, local_path)
research_agent/inno/environment/markdown_browser/mdconvert.py:646
↓ 2 callers
Method
_warp_args
(self, args_dict: str)
research_agent/inno/logger.py:18
↓ 2 callers
Function
abstract_composing
(research_field: str, instance_id: str)
paper_agent/abstract_composing.py:201
↓ 2 callers
Function
accuracy
Calculate accuracy between predictions and ground truth Args: y_pred (torch.Tensor): Predicted logits y_true (torch.Tensor):
examples/gnn_difformer/project/training/train.py:6
↓ 2 callers
Method
add_code_files
将指定目录下的所有代码文件添加到内存中。 Args: directory (str): 要添加的代码文件所在的目录路径
research_agent/inno/memory/codetree_memory.py:18
↓ 2 callers
Method
add_edges
(self, edges)
research_agent/inno/workflow/flowgraph.py:61
↓ 2 callers
Function
anonymize_text
Anonymize text content
benchmark_collection/1_create_inno_graph.py:273
↓ 2 callers
Method
calculate
Calculate inception score
examples/con_flowmatching/project/testing/scores.py:14
↓ 2 callers
Method
calculate
Calculate sample entropy score
examples/con_flowmatching/project/testing/scores.py:57
↓ 2 callers
Method
calculate_activation_statistics
Calculate activation statistics for FID score. Args: images: Batch of images Returns: tuple: Mean and
examples/fsq/project/testing/evaluator.py:20
↓ 2 callers
Method
calculate_activation_statistics
Calculate activation statistics (mean & covariance) for given images
examples/con_flowmatching/project/testing/evaluator.py:14
↓ 2 callers
Function
compress_folder
(source_folder, destination_folder, archive_name)
research_agent/inno/io_utils.py:25
↓ 2 callers
Function
conclusion_composing
(research_field: str, instance_id: str)
paper_agent/conclusion_composing.py:180
↓ 2 callers
Method
convert
(self, local_path, **kwargs)
research_agent/inno/environment/markdown_browser/mdconvert.py:438
↓ 2 callers
Function
convert_cookies_to_python
()
research_agent/inno/environment/browser_cookies.py:19
↓ 2 callers
Method
convert_local
(self, path: str, **kwargs: Any)
research_agent/inno/environment/markdown_browser/mdconvert.py:858
↓ 2 callers
Method
convert_response
( self, response: requests.Response, **kwargs: Any )
research_agent/inno/environment/mdconvert.py:917
↓ 2 callers
Method
convert_stream
(self, stream: Any, **kwargs: Any)
research_agent/inno/environment/mdconvert.py:878
↓ 2 callers
Function
decode_tokens_by_tiktoken
(tokens: list[int], model_name: str = "gpt-4o")
research_agent/inno/memory/utils.py:12
↓ 2 callers
Method
detailize_subsection
(self, structure, current_text, content, subsection)
paper_agent/methodology_composing_using_template.py:90
↓ 2 callers
Method
detailize_subsection
(self, structure, current_text, content, subsection=None)
paper_agent/introduction_composing.py:89
↓ 2 callers
Method
detailize_subsection
(self, structure: str, current_text: str, content: str, subsection: str)
paper_agent/experiments_composing.py:165
↓ 2 callers
Method
detailize_subsection
(self, structure: str, current_text: str, content: str, subsection: str)
paper_agent/related_work_composing_using_template.py:73
↓ 2 callers
Function
encode_string_by_tiktoken
(content: str, model_name: str = "gpt-4o")
research_agent/inno/memory/utils.py:4
↓ 2 callers
Function
execute_command
Execute a command in the system shell. Use this function when there is a need to run a system command, and execute programs. Args: co
research_agent/inno/tools/terminal_tools.py:265
↓ 2 callers
Function
experiments_composing
(research_field: str, instance_id: str)
paper_agent/experiments_composing.py:376
↓ 2 callers
Function
extract_answer_gsm8k
(completion)
benchmark/process/dataset_candidate/reasoning/math_reasoning/get_score.py:41
↓ 2 callers
Method
f_theta
Compute f_θ(t, x_t) = x_t + (1-t)v_θ(t, x_t)
examples/con_flowmatching/project/model/network.py:39
↓ 2 callers
Method
f_theta
Compute f_θ(t, x_t) = x_t + (1-t)v_θ(t, x_t)
examples/con_flowmatching/project/model/resnet_velocity.py:122
↓ 2 callers
Method
find_nearest_codebook
Find nearest codebook vector using rotated encoder output to improve gradient propagation.
examples/rotation_vq/project/model/vqvae.py:94
↓ 2 callers
Method
find_next
Scroll to the next viewport that matches the query
research_agent/inno/environment/markdown_browser/requests_markdown_browser.py:229
↓ 2 callers
Method
generate_or_revise_structure
(self, content, current_structure, iteration)
paper_agent/methodology_composing_using_template.py:26
↓ 2 callers
Method
generate_or_revise_structure
(self, content, current_structure, iteration)
paper_agent/introduction_composing.py:44
↓ 2 callers
Function
get_args
()
research_agent/run_infer_idea.py:60
↓ 2 callers
Function
get_args_research
()
main_ai_researcher.py:14
↓ 2 callers
Function
get_arxiv_paper_meta
get the meta data of the arxiv paper Args: arxiv_id: the arxiv id of the paper, can be the full url or the id (e.g. '2305.02759'
research_agent/inno/tools/inno_tools/paper_search.py:4
↓ 2 callers
Method
get_chat_completion_async
( self, agent: Agent, history: List, context_variables: dict, model_ov
research_agent/inno/core.py:344
↓ 2 callers
Function
get_code_survey_agent
(model: str, **kwargs)
research_agent/inno/agents/inno_agent/idea_agent.py:160
↓ 2 callers
Function
get_coding_plan_agent
(model: str, **kwargs)
research_agent/inno/agents/inno_agent/plan_agent.py:30
↓ 2 callers
Function
get_current_branch
f""" Get the current branch of the 'metachain'.
research_agent/inno/tools/github_ops.py:9
↓ 2 callers
Function
get_error_prefix
(last_browser_action: str, last_browser_action_error: str)
research_agent/inno/tools/inno_tools/web_tools.py:132
↓ 2 callers
Function
get_exp_analyser_agent
(model: str = "gpt-4o", **kwargs)
research_agent/inno/agents/inno_agent/exp_analyser.py:44
↓ 2 callers
Method
get_instance
(cls)
research_agent/inno/logger.py:121
↓ 2 callers
Function
get_judge_agent
(model: str, **kwargs)
research_agent/inno/agents/inno_agent/judge_agent.py:79
↓ 2 callers
Function
get_ml_agent
(model: str, **kwargs)
research_agent/inno/agents/inno_agent/ml_agent.py:28
↓ 2 callers
Function
get_paper_survey_agent
(model: str, **kwargs)
research_agent/inno/agents/inno_agent/survey_agent.py:21
↓ 2 callers
Function
get_prepare_agent
(model: str, **kwargs)
research_agent/inno/agents/inno_agent/prepare_agent.py:32
↓ 2 callers
Method
get_transformation_matrix
(self, embedding)
examples/hgcl/project/model/meta_network.py:22
↓ 2 callers
Function
introduction_composing
(research_field: str, instance_id: str)
paper_agent/introduction_composing.py:247
↓ 2 callers
Function
is_api_related
判断环境变量是否与API相关 Args: key: 环境变量名 Returns: bool: 是否与API相关
web_ai_researcher.py:728
↓ 2 callers
Function
load_cookies_from_json
(json_path)
research_agent/inno/environment/browser_cookies.py:6
↓ 2 callers
Function
load_env_vars
加载环境变量并返回字典格式 Returns: dict: 环境变量字典,每个值为一个包含值和来源的元组 (value, source)
web_ai_researcher.py:595
↓ 2 callers
Function
log_output
Log LLM interaction details
benchmark_collection/1_create_inno_graph.py:43
↓ 2 callers
Function
merge_fields
(target, source)
research_agent/inno/util.py:100
↓ 2 callers
Function
methodology_composing
(research_field: str, instance_id: str)
paper_agent/methodology_composing_using_template.py:314
↓ 2 callers
Function
normalize_adj_matrix
Symmetrically normalize adjacency matrix
examples/hgcl/project/data_processing/dataset.py:42
↓ 2 callers
Method
open_local_file
Convert a local file path to a file:/// URI, update the address, visit the page, and return the contents of the viewport.
research_agent/inno/environment/markdown_browser/requests_markdown_browser.py:302
↓ 2 callers
Method
page_down
Move the viewport down one page, if possible.
research_agent/inno/environment/markdown_browser/requests_markdown_browser.py:199
↓ 2 callers
Method
page_up
Move the viewport up one page, if possible.
research_agent/inno/environment/markdown_browser/requests_markdown_browser.py:203
↓ 2 callers
Method
peek
peek: peek the memery. Args: collection: the name of the collection to peek. n_results: the number of results
research_agent/inno/memory/rag_memory.py:117
↓ 2 callers
Function
process_interaction_data
Process interaction data and convert to matrix format
examples/hgcl/project/prepare_data.py:16
↓ 2 callers
Method
query_code
Query the code memory. Args: query_text (str): The query text n_results (int): The number of results
research_agent/inno/memory/codetree_memory.py:69
↓ 2 callers
Function
read_pdf
Interface function for PDF reading.
benchmark_collection/utils/pdf_utils.py:339
↓ 2 callers
Function
related_work_composing
(research_field: str, instance_id: str)
paper_agent/related_work_composing_using_template.py:275
↓ 2 callers
Method
rerank
(self, query_text: str, query_results: List[Dict])
research_agent/inno/memory/code_memory.py:90
↓ 2 callers
Method
run_comprehensive_evaluation
Run comprehensive model evaluation
examples/rotation_vq/project/testing/evaluator.py:100
↓ 2 callers
Function
run_experiment
Run a single experiment with enhanced model.
examples/gnn_nodeformer/project/experiments/run_enhanced_experiments.py:52
↓ 2 callers
Function
run_python
Run a python script. Args: code_path: The absolute or relative path (the relative path is from the root of the workplace `/workplace
research_agent/inno/tools/terminal_tools.py:290
↓ 2 callers
Function
run_sensitivity_experiment
Run sensitivity analysis experiment with given parameters
examples/con_flowmatching/project/experiments/sensitivity_analysis.py:23
↓ 2 callers
Method
save_cache
(self, agent_name, messages, iter_times: int = None, context_variables: Dict = None)
research_agent/inno/workflow/flowcache.py:46
↓ 2 callers
Method
search_code
搜索代码
research_agent/inno/tools/github_client.py:159
↓ 2 callers
Function
search_github_repos
Search GitHub public repositories based on a keyword. :param query: The query to search for in repository names or descriptions. :param
research_agent/inno/tools/code_search.py:8
← previous
next →
101–200 of 951, ranked by callers