Module Import alias Import alias identifier Import alias Import alias ImportFrom alias FunctionDef arguments args defaults body Assign NameStore Call AttributeLoad NameLoad attr Expr Call AttributeLoad NameLoad attr Str keyword NameLoad Expr Call AttributeLoad NameLoad attr Str keyword NameLoad keyword Num Expr Call AttributeLoad NameLoad attr Str keyword NameLoad keyword Num Expr Call AttributeLoad NameLoad attr Str keyword NameLoad keyword Num Expr Call AttributeLoad NameLoad attr Str keyword Str Return Call AttributeLoad NameLoad attr decorator_list FunctionDef arguments args NameParam NameParam NameParam defaults body Expr Str Assign TupleStore NameStore NameStore AttributeLoad NameLoad attr Assign NameStore SubscriptLoad AttributeLoad NameLoad attr Index TupleLoad Call AttributeLoad NameLoad attr TupleLoad NameLoad Num NameLoad NameLoad Assign NameStore Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad Assign NameStore Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad AugAssignAdd SubscriptStore NameLoad ExtSlice Slice BinOpAdd NameLoad Num Slice BinOpAdd NameLoad Num BinOpMult NameLoad NameLoad Assign NameStore IfExp Call AttributeLoad NameLoad attr NameLoad BinOpMult Call AttributeLoad NameLoad attr NameLoad NameLoad NameLoad AugAssignAdd SubscriptStore NameLoad ExtSlice Slice Num BinOpAdd NameLoad Num Slice Num BinOpAdd NameLoad Num Call AttributeLoad NameLoad attr NameLoad Assign NameStore AttributeLoad SubscriptLoad AttributeLoad NameLoad attr Index TupleLoad Call AttributeLoad NameLoad attr TupleLoad BinOpAdd NameLoad Num BinOpAdd NameLoad Num SubscriptLoad AttributeLoad NameLoad attr Index TupleLoad Call AttributeLoad NameLoad attr TupleLoad NameLoad Num NameLoad attr Assign TupleStore NameStore NameStore NameStore Call AttributeLoad AttributeLoad NameLoad attr attr AttributeLoad NameLoad attr Assign NameStore SubscriptLoad NameLoad ExtSlice Slice Slice NameLoad Assign TupleStore NameStore NameStore NameStore Call AttributeLoad AttributeLoad NameLoad attr attr Call AttributeLoad AttributeLoad NameLoad attr attr Call AttributeLoad NameLoad attr NameLoad Assign NameStore Dict Str Str Str Str Str NameLoad NameLoad NameLoad NameLoad NameLoad Return TupleLoad NameLoad NameLoad decorator_list FunctionDef arguments args defaults body Assign NameStore Call NameLoad Assign NameStore Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr Str Assign NameStore Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr AttributeLoad NameLoad attr AttributeLoad NameLoad attr For TupleStore NameStore NameStore Call AttributeLoad NameLoad attr body If CompareNotIn Str NameLoad body Assign NameStore Call AttributeLoad NameLoad attr Str orelse Assign NameStore SubscriptLoad NameLoad Index Str Assign NameStore SubscriptLoad SubscriptLoad NameLoad Index Str Slice For NameStore NameLoad body If CompareIn Call NameLoad NameLoad NameLoad body Continue Assign NameStore Call AttributeLoad NameLoad attr Call NameLoad NameLoad Assign TupleStore NameStore NameStore Call NameLoad NameLoad NameLoad For TupleStore NameStore NameStore Call AttributeLoad NameLoad attr body Assign SubscriptStore NameLoad Index NameLoad NameLoad If AttributeLoad NameLoad attr body Print Call AttributeLoad Str attr NameLoad NameLoad Expr Call AttributeLoad NameLoad attr decorator_list If CompareEq NameLoad Str body Expr Call NameLoad
N/A N/A h5py N/A numpy np N/A scipy.linalg N/A argparse lfd.rapprentice.tps tps_kernel_matrix parse_arguments N/A N/A N/A N/A N/A parser N/A N/A argparse ArgumentParser N/A N/A N/A parser add_argument datafile type str N/A N/A N/A parser add_argument --bend_coeff_init type float default 10 N/A N/A N/A parser add_argument --bend_coeff_final type float default 0.1 N/A N/A N/A parser add_argument --n_iter type int default 20 N/A N/A N/A parser add_argument --verbose action store_true N/A N/A N/A parser parse_args N/A get_lu_decomp N/A N/A x_na bend_coef rot_coef N/A N/A N/A 
    precomputes the LU decomposition and other intermediate results needed
    to fit a TPS to x_na with bend_coeff

    all thats needed is to compute the righthand side and do a forward solve
     N/A N/A n d N/A x_na shape N/A Q N/A N/A np c_ N/A N/A N/A N/A np ones N/A n 1 x_na K_nn N/A QWQ N/A N/A N/A Q T dot WQ N/A H N/A N/A N/A Q T dot Q N/A N/A H N/A N/A N/A d 1 N/A N/A d 1 N/A bend_coef K_nn N/A rot_coefs N/A N/A N/A np isscalar rot_coef N/A N/A N/A np ones d rot_coef rot_coef N/A N/A H N/A N/A 1 N/A d 1 N/A 1 N/A d 1 N/A N/A np diag rot_coefs N/A A N/A N/A N/A np r_ N/A N/A N/A N/A np zeros N/A N/A d 1 N/A d 1 N/A N/A np c_ N/A N/A N/A N/A np ones N/A n 1 x_na T N/A N/A _u _s _vh N/A N/A N/A np linalg svd N/A A T N/A N N/A _u N/A N/A N/A n_cnts N/A N/A p l u N/A N/A N/A scipy linalg lu N/A N/A N/A N T dot N/A N/A H dot N N/A res_dict N/A p l u N rot_coeffs p l u N rot_coeffs N/A N/A bend_coeff res_dict N/A main N/A N/A N/A N/A N/A args N/A parse_arguments N/A f N/A N/A h5py File N/A args datafile r N/A bend_coefs N/A N/A np loglinspace N/A args bend_coef_init N/A args bend_coeff_final N/A args n_iter N/A N/A seg_name seg_info N/A N/A f iteritems N/A N/A N/A LU seg_info N/A N/A lu_group N/A N/A seg_info create_group LU N/A N/A lu_group N/A seg_info N/A LU N/A x_na N/A N/A seg_info N/A cloud_xyz N/A N/A bend_coeff bend_coeffs N/A N/A N/A N/A str bend_coeff lu_group N/A N/A N/A bend_coeff_g N/A N/A lu_group create_group N/A str bend_coeff N/A N/A _ res N/A get_lu_decomp x_na bend_coeff N/A N/A k v N/A N/A res iteritems N/A N/A N/A bend_coeff_g N/A k v N/A N/A args verbose N/A N/A N/A N/A segment {}  bend_coeff {} format seg_name bend_coeff N/A N/A N/A f close N/A N/A N/A __name__ __main__ N/A N/A N/A main