Module Expr Str Import alias Import alias identifier ImportFrom alias alias FunctionDef arguments args defaults body Return BinOpMod Str NameLoad decorator_list FunctionDef arguments args defaults body Return BinOpMod Str TupleLoad NameLoad NameLoad decorator_list FunctionDef arguments args NameParam NameParam defaults Num Num kwarg body Assign TupleStore NameStore NameStore Call NameLoad NameLoad NameLoad Assign NameStore Call NameLoad keyword NameLoad keyword NameLoad Assign AttributeStore NameLoad attr Call NameLoad Return NameLoad decorator_list FunctionDef arguments args NameParam NameParam defaults Num Num kwarg body Assign TupleStore NameStore NameStore Call NameLoad NameLoad NameLoad Assign NameStore Call NameLoad keyword NameLoad keyword NameLoad Assign NameStore Call NameLoad NameLoad NameLoad Assign AttributeStore NameLoad attr Call NameLoad Return NameLoad decorator_list FunctionDef arguments args NameParam defaults body Expr Str Global identifier identifier Assign NameStore NameLoad Assign NameStore Num Global identifier Global identifier Global identifier Global identifier Assign NameStore Call AttributeLoad NameLoad attr NameLoad Assign NameStore BinOpDiv NameLoad Call AttributeLoad NameLoad attr NameLoad Assign NameStore Call AttributeLoad NameLoad attr TupleLoad NameLoad NameLoad Assign NameStore Call AttributeLoad NameLoad attr TupleLoad NameLoad NameLoad NameLoad Assign NameStore Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr For NameStore Call NameLoad NameLoad body Assign SubscriptStore NameLoad Index NameLoad Call NameLoad NameLoad NameLoad Assign SubscriptStore NameLoad Index NameLoad Call AttributeLoad AttributeLoad NameLoad attr attr SubscriptLoad NameLoad Index NameLoad decorator_list FunctionDef arguments args NameParam NameParam defaults body Expr Str Assign NameStore Call AttributeLoad NameLoad attr NameLoad Assign NameStore Call AttributeLoad NameLoad attr TupleLoad NameLoad NameLoad If CompareEq BinOpMod NameLoad Num Num body Assign SubscriptStore NameLoad Slice BinOpDiv NameLoad Num Num Assign NameStore Call AttributeLoad NameLoad attr NameLoad BinOpDiv NameLoad Num If CompareEq BinOpMod NameLoad Num Num body Assign SubscriptStore NameLoad Index Call AttributeLoad NameLoad attr NameLoad Num Assign NameStore Call AttributeLoad NameLoad attr NameLoad BinOpDiv BinOpSub NameLoad Num Num If CompareEq BinOpMod NameLoad Num Num body Assign SubscriptStore NameLoad Index Call AttributeLoad NameLoad attr NameLoad Num Num Assign NameStore Call AttributeLoad NameLoad attr NameLoad BinOpDiv BinOpSub NameLoad Num Num If CompareEq BinOpMod NameLoad Num Num body Assign SubscriptStore NameLoad Index Call AttributeLoad NameLoad attr NameLoad Num Num Assign NameStore Call AttributeLoad NameLoad attr NameLoad BinOpDiv BinOpSub NameLoad Num Num Return NameLoad decorator_list FunctionDef arguments args NameParam NameParam NameParam defaults Num body Expr Str Assign NameStore Call NameLoad NameLoad NameLoad Assign NameStore BinOpMult Num NameLoad Assign NameStore Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad Assign NameStore BinOpMult NameLoad Call AttributeLoad NameLoad attr NameLoad NameLoad Assign NameStore Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr TupleLoad NameLoad Num AugAssignMult NameStore Call AttributeLoad NameLoad attr NameLoad Num Assign NameStore BinOpAdd NameLoad NameLoad Assign NameStore Call AttributeLoad NameLoad attr AttributeLoad NameLoad attr Return BinOpAdd NameLoad BinOpMult Num Call AttributeLoad NameLoad attr NameLoad decorator_list FunctionDef arguments args NameParam NameParam defaults body Assign NameStore Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad Return BinOpAdd SubscriptLoad NameLoad ExtSlice Index NameLoad Slice AttributeLoad Call AttributeLoad NameLoad attr AttributeLoad SubscriptLoad NameLoad Index NameLoad attr Call AttributeLoad NameLoad attr NameLoad NameLoad attr decorator_list FunctionDef arguments args NameParam NameParam defaults body Assign NameStore Call AttributeLoad AttributeLoad NameLoad attr attr NameLoad Assign NameStore Call NameLoad Assign NameStore Call AttributeLoad NameLoad attr NameLoad NameLoad Assign NameStore Call NameLoad For NameStore Call NameLoad NameLoad body Expr Call AttributeLoad NameLoad attr Call NameLoad NameLoad SubscriptLoad NameLoad Index NameLoad Expr Call AttributeLoad NameLoad attr BinOpMult NameLoad Call AttributeLoad NameLoad attr SubscriptLoad NameLoad Index NameLoad Assign NameStore Call AttributeLoad NameLoad attr NameLoad Assign NameStore Call AttributeLoad NameLoad attr NameLoad Assign NameStore Call AttributeLoad NameLoad attr SubscriptLoad AttributeLoad NameLoad attr Index Num Assign NameStore SubscriptLoad NameLoad Index NameLoad Assign NameStore SubscriptLoad NameLoad Index NameLoad Return TupleLoad NameLoad NameLoad decorator_list FunctionDef arguments args NameParam defaults NameLoad body ImportFrom alias Expr Call AttributeLoad NameLoad attr For NameStore Call NameLoad NameLoad body Expr Call AttributeLoad NameLoad attr Num Num BinOpAdd NameLoad Num Assign NameStore Call NameLoad NameLoad NameLoad Expr Call AttributeLoad NameLoad attr NameLoad keyword Str If NameLoad body Expr Call AttributeLoad NameLoad attr decorator_list FunctionDef arguments args NameParam defaults NameLoad body ImportFrom alias Expr Call AttributeLoad NameLoad attr For NameStore Call NameLoad NameLoad body Expr Call AttributeLoad NameLoad attr Num Num BinOpAdd NameLoad Num Expr Call AttributeLoad NameLoad attr SubscriptLoad NameLoad Index NameLoad keyword Str If NameLoad body Expr Call AttributeLoad NameLoad attr decorator_list
N/A N/A 
DeadLeaves.py

 N/A scipy.linalg N/A numpy np bnpy.data XData MinibatchIterator get_short_name N/A N/A N/A N/A N/A N/A DeadLeavesD%d D N/A get_data_info N/A N/A N/A N/A N/A N/A Dead Leaves Data. K=%d. D=%d. N/A K D N/A get_data N/A N/A seed nObsTotal N/A 8675309 25000 kwargs N/A N/A N/A X TrueZ N/A generateData seed nObsTotal N/A Data N/A XData X X TrueZ TrueZ N/A N/A Data summary N/A get_data_info N/A Data N/A get_minibatch_iterator N/A N/A seed nObsTotal N/A 8675309 25000 kwargs N/A N/A N/A X TrueZ N/A generateData seed nObsTotal N/A Data N/A XData X X TrueZ TrueZ N/A DataIterator N/A MinibatchIterator Data kwargs N/A N/A DataIterator summary N/A get_data_info N/A DataIterator N/A makeTrueParams N/A N/A Din N/A N/A N/A  Create mixture weights, means, covariance matrices for all components
   N/A K D N/A D Din N/A K 8 N/A w N/A Mu N/A Sigma N/A cholSigma N/A w N/A N/A np ones K N/A w N/A w N/A N/A np sum w N/A Mu N/A N/A np zeros N/A K D N/A Sigma N/A N/A np zeros N/A K D D N/A cholSigma N/A N/A np zeros N/A Sigma shape N/A k N/A xrange K N/A N/A N/A Sigma N/A k N/A makeImgPatchCovMatForComp D k N/A N/A cholSigma N/A k N/A N/A N/A scipy linalg cholesky N/A Sigma N/A k N/A makeImgPatchPrototype N/A N/A D compID N/A N/A N/A  Create image patch prototype for specific component
      Returns
      --------
      Xprototype : sqrt(D) x sqrt(D) matrix
   N/A P N/A N/A np sqrt D N/A Xprototype N/A N/A np zeros N/A P P N/A N/A N/A compID 4 0 N/A N/A N/A Xprototype N/A N/A P 2 1.0 N/A Xprototype N/A N/A np rot90 Xprototype N/A compID 4 N/A N/A N/A compID 4 1 N/A N/A N/A Xprototype N/A N/A N/A np tril_indices P 1 N/A Xprototype N/A N/A np rot90 Xprototype N/A N/A compID 1 4 N/A N/A N/A compID 4 2 N/A N/A N/A Xprototype N/A N/A N/A np tril_indices P 2 1 N/A Xprototype N/A N/A np rot90 Xprototype N/A N/A compID 2 4 N/A N/A N/A compID 4 3 N/A N/A N/A Xprototype N/A N/A N/A np tril_indices P -2 1 N/A Xprototype N/A N/A np rot90 Xprototype N/A N/A compID 3 4 N/A Xprototype N/A makeImgPatchCovMatForComp N/A N/A D compID sig N/A 0.005 N/A N/A  Create img patch covariance matrix for specific component

      Returns
      --------
      Sigma : D x D covariance matrix
   N/A Xprototype N/A makeImgPatchPrototype D compID N/A N N/A 100 D N/A PRNG N/A N/A N/A np random RandomState compID N/A Xnoise N/A sig N/A N/A PRNG randn N D N/A Xsignal N/A N/A np tile N/A N/A Xprototype flatten N/A N 1 N/A Xsignal N/A N/A PRNG randn N 1 N/A X N/A Xnoise Xsignal N/A Sigma N/A N/A np cov N/A X T N/A N/A Sigma N/A 1e-05 N/A N/A np eye D N/A sample_data_from_comp N/A N/A k Nk N/A N/A N/A PRNG N/A N/A N/A np random RandomState k N/A N/A N/A Mu N/A N/A k N/A N/A N/A N/A np dot N/A N/A cholSigma N/A k T N/A N/A PRNG randn D Nk T N/A generateData N/A N/A seed nObsTotal N/A N/A N/A PRNG N/A N/A N/A np random RandomState seed N/A trueList N/A list N/A Npercomp N/A N/A PRNG multinomial nObsTotal w N/A X N/A list N/A k N/A range K N/A N/A N/A N/A X append N/A sample_data_from_comp k N/A Npercomp N/A k N/A N/A N/A trueList append N/A k N/A N/A np ones N/A Npercomp N/A k N/A X N/A N/A np vstack X N/A TrueZ N/A N/A np hstack trueList N/A permIDs N/A N/A PRNG permutation N/A N/A X shape N/A 0 N/A X N/A X N/A permIDs N/A TrueZ N/A TrueZ N/A permIDs N/A N/A X TrueZ N/A plotImgPatchPrototypes N/A N/A doShowNow N/A True N/A matplotlib pylab N/A N/A N/A pylab figure N/A kk N/A range K N/A N/A N/A N/A pylab subplot 2 4 N/A kk 1 N/A Xp N/A makeImgPatchPrototype D kk N/A N/A N/A pylab imshow Xp interpolation nearest N/A doShowNow N/A N/A N/A N/A pylab show N/A plotTrueCovMats N/A N/A doShowNow N/A True N/A matplotlib pylab N/A N/A N/A pylab figure N/A kk N/A range K N/A N/A N/A N/A pylab subplot 2 4 N/A kk 1 N/A N/A N/A pylab imshow N/A Sigma N/A kk interpolation nearest N/A doShowNow N/A N/A N/A N/A pylab show N/A