Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/SakanaAI/AI-Scientist-v2
/ types & classes
Types & classes
36 in github.com/SakanaAI/AI-Scientist-v2
⨍
Functions
307
◇
Types & classes
36
↓ 9 callers
Class
FunctionSpec
ai_scientist/treesearch/backend/utils.py:106
↓ 7 callers
Class
Node
A single node in the solution tree. Contains code, execution results, and evaluation information.
ai_scientist/treesearch/journal.py:44
↓ 4 callers
Class
Journal
A collection of nodes representing the solution tree.
ai_scientist/treesearch/journal.py:362
↓ 4 callers
Class
WorstMetricValue
Represents an invalid metric value, e.g. when the agent creates a buggy solution. Always compares worse than any valid metric value.
ai_scientist/treesearch/utils/metric.py:328
↓ 3 callers
Class
HuggingFaceImageNet
ai_scientist/ideas/i_cant_believe_its_not_betterrealworld.py:143
↓ 3 callers
Class
HuggingFaceImageNet
ai_scientist/ideas/i_cant_believe_its_not_better.py:175
↓ 3 callers
Class
MetricValue
Represents one or more metric values to be optimized, which can be compared to other metric values. Comparisons (and max, min) are based on w
ai_scientist/treesearch/utils/metric.py:114
↓ 3 callers
Class
Stage
ai_scientist/treesearch/agent_manager.py:104
↓ 2 callers
Class
AblationIdea
Ablation idea
ai_scientist/treesearch/parallel_agent.py:238
↓ 2 callers
Class
HyperparamTuningIdea
Hyperparameter tuning idea
ai_scientist/treesearch/parallel_agent.py:246
↓ 2 callers
Class
Interpreter
ai_scientist/treesearch/interpreter.py:81
↓ 2 callers
Class
StageTransition
Records transition between stages and the reasoning
ai_scientist/treesearch/agent_manager.py:114
↓ 1 callers
Class
AgentManager
ai_scientist/treesearch/agent_manager.py:123
↓ 1 callers
Class
ExecutionResult
Result of executing a code snippet in the interpreter. Contains the output, execution time, and exception information.
ai_scientist/treesearch/interpreter.py:27
↓ 1 callers
Class
GPUManager
Manages GPU allocation across processes
ai_scientist/treesearch/parallel_agent.py:1091
↓ 1 callers
Class
MinimalAgent
A minimal agent class that only contains what's needed for processing nodes
ai_scientist/treesearch/parallel_agent.py:254
↓ 1 callers
Class
ParallelAgent
ai_scientist/treesearch/parallel_agent.py:1142
↓ 1 callers
Class
RedirectQueue
ai_scientist/treesearch/interpreter.py:70
↓ 1 callers
Class
SemanticScholarSearchTool
ai_scientist/tools/semantic_scholar.py:19
↓ 1 callers
Class
TokenTracker
ai_scientist/utils/token_tracker.py:10
Class
ai_scientist/treesearch/utils/viz_templates/template.js:35
Class
ai_scientist/treesearch/utils/viz_templates/template.js:138
Class
AblationConfig
Track state of ablation experiments
ai_scientist/treesearch/parallel_agent.py:223
Class
AgentConfig
ai_scientist/treesearch/utils/config.py:57
Class
BaseTool
An abstract base class for defining custom tools. Attributes: ----------- - name (str): The name of the tool. - description (str
ai_scientist/tools/base_tool.py:5
Class
Config
ai_scientist/treesearch/utils/config.py:89
Class
DebugConfig
ai_scientist/treesearch/utils/config.py:52
Class
Edge
ai_scientist/treesearch/utils/viz_templates/template.js:138
Class
ExecConfig
ai_scientist/treesearch/utils/config.py:77
Class
ExperimentConfig
ai_scientist/treesearch/utils/config.py:84
Class
InteractiveSession
A collection of nodes for an interaction session (when the agent interacts with a Jupyter notebook-like interface).
ai_scientist/treesearch/journal.py:332
Class
MetricValue_old
Represents the value of a metric to be optimized, which can be compared to other metric values. Comparisons (and max, min) are based on which
ai_scientist/treesearch/utils/metric.py:11
Class
Node
ai_scientist/treesearch/utils/viz_templates/template.js:35
Class
SearchConfig
ai_scientist/treesearch/utils/config.py:45
Class
StageConfig
ai_scientist/treesearch/utils/config.py:36
Class
ThinkingConfig
ai_scientist/treesearch/utils/config.py:30