MCPcopy Create free account

hub / github.com/SymposiumOrganization/NeuralSymbolicRegressionThatScales / functions

Functions173 in github.com/SymposiumOrganization/NeuralSymbolicRegressionThatScales

Method__getitem__
(self, index)
src/nesymres/architectures/data.py:65
Method__init__
(self,base_path: Path = None, target_path: Path = None, metadata=None)
src/nesymres/utils.py:15
Method__init__
(self, env: generator.Generator, is_timer=False)
src/nesymres/dataset/convert_csv_to_dataload_format.py:29
Method__init__
(self, params)
src/nesymres/dataset/generator.py:119
Method__init__
(self, fun, stop_after=10)
src/nesymres/architectures/bfgs.py:29
Method__init__
Initialize n-best list of hypotheses.
src/nesymres/architectures/beam_search.py:185
Method__init__
(self,cfg)
src/nesymres/architectures/set_encoder.py:7
Method__init__
( self, data_path: Path, cfg, mode: str )
src/nesymres/architectures/data.py:46
Method__init__
( self, data_train_path, data_val_path, data_test_path, cfg )
src/nesymres/architectures/data.py:239
Method__init__
(self, dim_in, dim_out, num_heads, ln=False)
src/nesymres/architectures/set_transformer.py:36
Method__init__
(self, dim_in, dim_out, num_heads, num_inds, ln=False)
src/nesymres/architectures/set_transformer.py:44
Method__init__
(self, dim, num_heads, num_seeds, ln=False)
src/nesymres/architectures/set_transformer.py:56
Method__init__
( self, cfg )
src/nesymres/architectures/model.py:17
Method__init__
(self, env: generator.Generator, is_timer=False)
scripts/csv_handling/convert_csv_to_dataload_format.py:29
Method__init__
(self, data_path, metadata, support, target_image: list, validation_eqs: set)
scripts/data_creation/filter_from_already_existing.py:34
Method__init__
(self, env: generator.Generator, number_of_equations, eq_per_block, h5_creator:H5FilesCreator, is_timer=False
scripts/data_creation/dataset_creation.py:27
Method__init__
Args: param1: param2: support: target_image: target_image_l: vali
scripts/data_creation/test_presence.py:35
Method__len__
Number of hypotheses in the list.
src/nesymres/architectures/beam_search.py:196
Method__len__
(self)
src/nesymres/architectures/data.py:94
Method_prefix_to_edges
(self, expr)
src/nesymres/dataset/generator.py:519
Functionbfgs
(pred_str, X, y, cfg)
src/nesymres/architectures/bfgs.py:44
Functioncheck_additive_constants
Extract a non-constant sub-tree from an equation.
src/nesymres/dataset/sympy_utils.py:188
Functionclean_degree2_solution
Clean solutions of second order differential equations.
src/nesymres/dataset/sympy_utils.py:357
Functioncode_pickler
(code)
src/nesymres/utils.py:48
Functioncode_unpickler
(data)
src/nesymres/utils.py:45
Methodconfigure_optimizers
(self)
src/nesymres/architectures/model.py:123
Functioncount_parameters
(model)
src/nesymres/dataset/data_utils.py:75
Methodcreate_block
(self,block_idx)
scripts/data_creation/dataset_creation.py:37
Functioncustom_collate_fn
(eqs: List[Equation], cfg)
src/nesymres/architectures/data.py:97
Functiondataset_loader
(train_dataset, test_dataset, batch_size=1024, valid_size=0.20)
src/nesymres/dataset/data_utils.py:38
Functionde_tokenize
(tokenized_expr, id2word:dict)
src/nesymres/architectures/data.py:121
Functionepoch_time
(start_time, end_time)
src/nesymres/dataset/data_utils.py:79
Functionevaluate_equation
Evaluate equation `pred_equation` given as a string.
scripts/benchmark/fit.py:61
Functionevaluate_model
model_predict is a callable that takes an X of shape (n_datapoints, n_variables) and returns scalar predictions
src/nesymres/benchmark.py:10
Functionevaluate_sklearn
Evaluate sklearn model at model_path.
scripts/benchmark/fit.py:76
Functionevaluate_sklearn
Evaluate sklearn model at model_path.
scripts/benchmark/collect_results.py:136
Methodfitfunc
Same API as fit functions in sklearn: X [Number_of_points, Number_of_features], Y [Number_of_points]
src/nesymres/architectures/model.py:128
Methodforward
(self, x)
src/nesymres/architectures/set_encoder.py:59
Methodforward
(self, Q, K)
src/nesymres/architectures/set_transformer.py:19
Methodforward
(self, X)
src/nesymres/architectures/set_transformer.py:40
Methodforward
(self, X)
src/nesymres/architectures/set_transformer.py:51
Methodforward
(self, X)
src/nesymres/architectures/set_transformer.py:62
Methodfun
(self, x, *args)
src/nesymres/architectures/bfgs.py:34
Functiongenerate_beam
Decode a sentence given initial start. `x`: - LongTensor(bs, slen) <EOS> W1 W2 W3 <EOS> <PAD> <EOS> W1 W2 W3
src/nesymres/architectures/beam_search.py:3
Functionget_data_reject_nan
(eq: Equation, cfg)
src/nesymres/benchmark.py:65
Functiongroup_symbolically_indetical_eqs
(data,indexes_dict,disjoint_sets)
src/nesymres/dataset/data_utils.py:27
Methodhandler
(self,signum, frame)
src/nesymres/dataset/convert_csv_to_dataload_format.py:36
Methodhandler
(self,signum, frame)
scripts/csv_handling/convert_csv_to_dataload_format.py:36
Methodhandler
(self,signum, frame)
scripts/data_creation/dataset_creation.py:48
Functionhas_I
Detect whether some expressions contain complex numbers.
src/nesymres/dataset/sympy_utils.py:399
Functionhas_inf_nan
Detect whether some expressions contain a NaN / Infinity symbol.
src/nesymres/dataset/sympy_utils.py:389
Functioninit_weights
(m)
src/nesymres/dataset/data_utils.py:64
Functioninitialize_weights
Used for the transformer
src/nesymres/dataset/data_utils.py:69
Methodis_valid_and_not_in_validation_set
Return True if an equation is not the validation set and is numerically meaningfull (i.e. values all differs from nan, -inf, +inf, all zeros)
scripts/data_creation/filter_from_already_existing.py:43
Methodis_valid_and_not_in_validation_set
Assert both symbolically and numerically that the equation is not in the validation set Args: idx: index to the Eq in the
scripts/data_creation/test_presence.py:55
Functionload_equation
(benchmark_path, equation_idx)
src/nesymres/benchmark.py:139
Methodmake_src_mask
(self, src)
src/nesymres/architectures/model.py:57
Functionparse_args
()
scripts/benchmark/collect_results.py:20
Functionpatch_benchmark_name
For experiments before 2021-02-03, there was no key 'benchmark_name'. They were all using the AI Feynman dataset
scripts/benchmark/collect_results.py:48
Functionplot_durations
(combined_df)
scripts/benchmark/collect_results.py:211
Methodrecreate_single_hd5_from_idx
(self,block:Tuple)
src/nesymres/utils.py:35
Functionreindex_coefficients
Re-index coefficients (i.e. if a1 is there and not a0, replace a1 by a0, and recursively).
src/nesymres/dataset/sympy_utils.py:250
Functionremove_mul_const
Remove the multiplicative factor of an expression, and return it.
src/nesymres/dataset/sympy_utils.py:133
Functionrename_benchmark
We updated our benchmark name from "ours" to "ours-nc".
scripts/benchmark/collect_results.py:28
Methodreturn_training_set
(self, i)
src/nesymres/dataset/convert_csv_to_dataload_format.py:39
Methodreturn_training_set
(self, i)
scripts/csv_handling/convert_csv_to_dataload_format.py:39
Functionsimplify_equa_diff
Simplify a differential equation by removing non-zero factors.
src/nesymres/dataset/sympy_utils.py:313
Functionsmallest_with_symbols
Return the smallest sub-tree in an expression that contains all given symbols.
src/nesymres/dataset/sympy_utils.py:330
Functionstandardize_equation
Replace x1 -> x, x2 -> y, x3 -> z in order to work with equivalent formats. Also, add 0 times all variables in order to make all variables ap
scripts/benchmark/collect_results.py:35
Methodtrain_dataloader
returns training dataloader
src/nesymres/architectures/data.py:277
Methodtraining_step
(self, batch, _)
src/nesymres/architectures/model.py:112
Methodval_dataloader
returns validation dataloader
src/nesymres/architectures/data.py:290
Methodvalidation_step
(self, batch, _)
src/nesymres/architectures/model.py:118
← previous101–173 of 173, ranked by callers