MCPcopy Create free account

hub / github.com/NewT123-WM/tnlearn / functions

Functions330 in github.com/NewT123-WM/tnlearn

Method__init__
(self, decoder_layer: TNTransformerDecoderLayer, num_layers: int, norm: Optional[LayerNorm] = None)
tnlearn/modules/TNtransformer.py:537
Method__init__
(self, mode: str, input_size: int, hidden_size: int,
tnlearn/modules/TNrnn.py:140
Method__init__
(self, input_size: int, hidden_size: int, num_layers: int = 1, bias: bool = True, batch_first
tnlearn/modules/TNrnn.py:280
Method__init__
(self, input_size: int, hidden_size: int, num_layers: int = 1, bias: bool = True, batch_first
tnlearn/modules/TNrnn.py:308
Method__init__
(self, input_size: int, hidden_size: int, bias: bool, symbolic_expression: str = 'x', num_chu
tnlearn/modules/TNrnn.py:339
Method__init__
(self, input_size: int, hidden_size: int, bias: bool = True, nonlinearity: str = 'tanh', symb
tnlearn/modules/TNrnn.py:398
Method__init__
(self, input_size: int, hidden_size: int, bias: bool = True, symbolic_expression: str = 'x',
tnlearn/modules/TNrnn.py:456
Method__init__
(self, input_size: int, hidden_size: int, bias: bool = True, symbolic_expression: str = 'x',
tnlearn/modules/TNrnn.py:522
Method__init__
(self, api_key: str, model: str, base_url: str = "https://api.deepseek.com", verbose: Union[bool, int] = False
tnlearn/drsr/llm.py:241
Method__init__
(self, api_key: str, model: str, base_url: str = "https://api.siliconflow.cn/v1", verbose: Union[bool, int] =
tnlearn/drsr/llm.py:246
Method__init__
(self, api_key: str, model: str, base_url: str = "https://uni-api.cstcloud.cn/v1", verbose: Union[bool, int] =
tnlearn/drsr/llm.py:257
Method__init__
(self, api_key: str, model: str, base_url: str = "http://localhost:11111/v1", verbose: Union[bool, int] = Fals
tnlearn/drsr/llm.py:266
Method__init__
(self, api_key: str, model: str, base_url: str = None, verbose: Union[bool, int] = False)
tnlearn/drsr/llm.py:275
Method__init__
(self, verbose=False, numba_accelerate=False, use_multiprocessing=None)
tnlearn/drsr/evaluator.py:47
Method__init__
( self, database: buffer.ExperienceBuffer, function_to_evolve: str, function_t
tnlearn/drsr/evaluator.py:153
Method__init__
( self, llm_config: Dict[str, Any], max_iterations: int = 20, samples_per_iter
tnlearn/drsr/agent.py:34
Method__init__
(self, sourcecode: str)
tnlearn/drsr/code_manipulation.py:127
Method__init__
(self, samples_per_prompt: int, client=None, trim: bool = True)
tnlearn/drsr/sampler.py:39
Method__init__
( self, database: buffer.ExperienceBuffer, evaluators: Sequence[evaluator.Evaluator],
tnlearn/drsr/sampler.py:112
Method__init__
( self, log_dir: str | None = None, pkl_dir: str | None = None,
tnlearn/drsr/profile.py:31
Method__init__
(self, score: float, implementation: str)
tnlearn/drsr/buffer.py:65
Method__init__
( self, functions_per_prompt: int, cluster_sampling_temperature_init: float, c
tnlearn/drsr/buffer.py:87
Method__init__
( self, config: config_lib.ExperienceBufferConfig, )
tnlearn/drsr/buffer.py:160
Method__init__
(self)
tests/test_base.py:23
Method__len__
(self)
tnlearn/utils.py:32
Method__setattr__
(self, name: str, value: str)
tnlearn/drsr/code_manipulation.py:72
Method__setstate__
(self, state)
tnlearn/modules/TNconv.py:89
Method__setstate__
(self, state)
tnlearn/modules/TNconv.py:314
Method__setstate__
(self, state)
tnlearn/modules/TNlinear.py:75
Method__setstate__
(self, state)
tnlearn/modules/TNtransformer.py:343
Method__setstate__
(self, state)
tnlearn/modules/TNrnn.py:223
Method__str__
(self)
tnlearn/drsr/code_manipulation.py:60
Method__str__
(self)
tnlearn/drsr/code_manipulation.py:93
Function_augment_input
Applies all basis functions to the input and concatenates the results. Args: x (Tensor): Input tensor of shape (..., feature_dim).
tnlearn/modules/TNrnn.py:95
Method_compile_and_run_function
(self, program, function_to_run, function_to_evolve, dataset, numba_accelera
tnlearn/drsr/evaluator.py:115
Method_conv_forward
(self, x, weight, padding)
tnlearn/modules/TNconv.py:200
Method_conv_forward
(self, x, weight, padding)
tnlearn/modules/TNconv.py:216
Method_conv_forward
(self, x, weight, padding)
tnlearn/modules/TNconv.py:232
Method_conv_forward
(self, x, weight)
tnlearn/modules/TNconv.py:408
Method_conv_forward
(self, x, weight)
tnlearn/modules/TNconv.py:425
Method_conv_forward
(self, x, weight)
tnlearn/modules/TNconv.py:442
Method_get_case_insensitive
(d: dict, k: str)
tnlearn/drsr/llm.py:317
Method_sort_key
(item)
tnlearn/drsr/profile.py:142
Methodbuild_model
(self, input_dim, output_dim)
tests/test_base.py:33
Methodcalculate_auc
r"""Calculate the Area Under the Receiver Operating Characteristic Curve (ROC AUC)
tnlearn/base1.py:81
Methodcalculate_f1_score
r"""Calculate the F1 score, a weighted average of precision and recall
tnlearn/base1.py:85
Methodcalculate_precision
r"""Calculate the precision, the ability of the classifier not to label a sample as positive if it is negative
tnlearn/base1.py:93
Methodcalculate_recall
r"""Calculate the recall, the ability of the classifier to find all the positive samples
tnlearn/base1.py:89
Methodcount_param
r"""Print the network structure and output the number of network parameters.
tnlearn/mlpclassifier.py:387
Methodcount_param
r"""Print the network structure and output the number of network parameters.
tnlearn/mlpregressor.py:355
Functiondensenet161
()
benchmark/image_benchmark_polyregressor/TC_densenet.py:130
Functiondensenet169
()
benchmark/image_benchmark_polyregressor/TC_densenet.py:124
Functiondensenet201
()
benchmark/image_benchmark_polyregressor/TC_densenet.py:127
Methoddraw_samples
(self, prompt: str, config: config_lib.Config, best_score: float = None, progress: float = 0.0, extra_prompt:
tnlearn/drsr/sampler.py:44
Methodextra_repr
(self)
tnlearn/modules/TNrnn.py:201
Methodextra_repr
(self)
tnlearn/modules/TNrnn.py:373
Methodfit
r"""Train the network with training data. Args: X (numpy ndarray): Training data. y (numpy ndarray): Label data.
tnlearn/mlpclassifier.py:199
Methodfit
(self, X, y, view_training_process=False)
tnlearn/poly_regressor.py:252
Methodfit
r"""Train the network with training data. Args: X (numpy ndarray): Training data. y (numpy ndarray): Label data.
tnlearn/mlpregressor.py:202
Methodfit
r"""The symbolic regression algorithm is performed to fit the data. Args: X: Input data. y: Label.
tnlearn/regressor.py:277
Methodfit
(self, X: np.ndarray, y: np.ndarray)
tnlearn/drsr/agent.py:131
Methodforward
(self, x)
benchmark/image_benchmark_polyregressor/TC_senet.py:50
Methodforward
(self, x)
benchmark/image_benchmark_polyregressor/TC_senet.py:100
Methodforward
(self, x)
benchmark/image_benchmark_polyregressor/TC_senet.py:135
Methodforward
(self, x)
benchmark/image_benchmark_polyregressor/TC_Googlenet.py:65
Methodforward
(self, x)
benchmark/image_benchmark_polyregressor/TC_Googlenet.py:111
Methodforward
(self, x)
benchmark/image_benchmark_polyregressor/TC_densenet.py:44
Methodforward
(self, x)
benchmark/image_benchmark_polyregressor/TC_densenet.py:62
Methodforward
(self, x)
benchmark/image_benchmark_polyregressor/TC_densenet.py:106
Methodforward
(self, x)
benchmark/image_benchmark_polyregressor/TC_Resnet.py:57
Methodforward
(self, x)
benchmark/image_benchmark_polyregressor/TC_Resnet.py:91
Methodforward
(self, x)
benchmark/image_benchmark_polyregressor/TC_Resnet.py:140
Methodforward
(self, state: torch.Tensor)
tnlearn/rl_regressor.py:56
Methodforward
前向传播:对每个编译好的子表达式计算张量,再分别做线性变换,最后求和加偏置。
tnlearn/neurons.py:123
Methodforward
(self, x)
tnlearn/TN_base.py:68
Methodforward
(self, x)
tnlearn/modules/TNconv.py:154
Methodforward
(self, x)
tnlearn/modules/TNconv.py:378
Methodforward
r"""Forward pass: single linear transformation after feature augmentation, plus bias.
tnlearn/modules/TNlinear.py:130
Methodforward
( self, src: Tensor, src_mask: Optional[Tensor] = None, src_ke
tnlearn/modules/TNtransformer.py:168
Methodforward
( self, tgt: Tensor, memory: Tensor, tgt_mask: Optional[Tensor
tnlearn/modules/TNtransformer.py:348
Methodforward
( self, src: Tensor, mask: Optional[Tensor] = None, src_key_pa
tnlearn/modules/TNtransformer.py:438
Methodforward
( self, tgt: Tensor, memory: Tensor, tgt_mask: Optional[Tensor
tnlearn/modules/TNtransformer.py:544
Methodforward
( self, src: Tensor, tgt: Tensor, src_mask: Optional[Tensor] =
tnlearn/modules/TNtransformer.py:646
Methodforward
Forward pass of the RNN. Args: input (Tensor): Input tensor. Shape depends on batch_first. hx (Tensor, optio
tnlearn/modules/TNrnn.py:187
Methodforward
Forward pass of the RNN cell. Args: input (Tensor): Input tensor of shape (batch, input_size) or (input_size,).
tnlearn/modules/TNrnn.py:405
Methodforward
Forward pass of the LSTM cell. Args: input (Tensor): Input tensor of shape (batch, input_size) or (input_size,).
tnlearn/modules/TNrnn.py:461
Methodforward
Forward pass of the GRU cell. Args: input (Tensor): Input tensor of shape (batch, input_size) or (input_size,).
tnlearn/modules/TNrnn.py:527
Methodgenerate_square_subsequent_mask
( sz: int, device: torch.device = torch.device(torch._C._get_default_device()),
tnlearn/modules/TNtransformer.py:678
Methodget_function
(self, function_name: str)
tnlearn/drsr/code_manipulation.py:118
Functionget_functions_called
Return the set of all functions called in `code`.
tnlearn/drsr/code_manipulation.py:270
Functionget_global_tokens
获取本次实验的全局 token 统计(thinking/content/total)。
tnlearn/drsr/llm.py:40
Methodget_prompt
(self)
tnlearn/drsr/buffer.py:185
Methodget_tensors
(self)
tnlearn/utils.py:36
Methodload
r"""Load a model from the specified path with the given filename. Args: path: The location of the trained model. file
tnlearn/base1.py:61
Functionmain
( specification: str, inputs: Sequence[Any], config: config_lib.Config, max_sample_nums: Optio
tnlearn/drsr/pipeline.py:40
Methodmultivariate_predict
(X)
tnlearn/drsr/agent.py:274
Methodnode_count
r"""Method to count the number of nodes in a given tree structure. Args: x: A symbol tree structure Returns:
tnlearn/regressor.py:245
Methodparse
(x)
tnlearn/modules/TNconv.py:326
Methodpredict
r"""Use a trained model to make predictions. Args: X (torch.Tensor): Data that needs to be predicted. Returns:
tnlearn/mlpclassifier.py:313
Methodpredict
(self, X)
tnlearn/poly_regressor.py:271
← previousnext →201–300 of 330, ranked by callers