Module Import alias identifier Import alias identifier ImportFrom alias ImportFrom alias FunctionDef arguments args NameParam defaults Str body Assign NameStore Call AttributeLoad NameLoad attr Call AttributeLoad NameLoad attr ListLoad ListLoad Num Num ListLoad Num Num ListLoad Num Num ListLoad Num Num ListLoad Num Num keyword AttributeLoad NameLoad attr Assign NameStore Call AttributeLoad NameLoad attr ListLoad Num Num Num Num Num keyword AttributeLoad NameLoad attr Assign NameStore Call AttributeLoad NameLoad attr ListLoad ListLoad Num Num ListLoad Num Num keyword AttributeLoad NameLoad attr Assign NameStore Call AttributeLoad NameLoad attr ListLoad Num Num keyword AttributeLoad NameLoad attr Assign NameStore Num If CompareEq NameLoad Str body Assign NameStore Call AttributeLoad NameLoad attr NameLoad Assign SubscriptStore NameLoad Index CompareLt NameLoad Call AttributeLoad NameLoad attr NameLoad Num Assign NameStore NameLoad Return TupleLoad NameLoad NameLoad NameLoad NameLoad NameLoad decorator_list FunctionDef arguments args defaults body Assign TupleStore NameStore NameStore NameStore NameStore NameStore Call NameLoad keyword Str Assign NameStore Call AttributeLoad NameLoad attr keyword Num keyword Num keyword Num keyword Num keyword Num Expr Call AttributeLoad NameLoad attr NameLoad NameLoad Assign NameStore Call AttributeLoad NameLoad attr NameLoad Assign NameStore Call AttributeLoad NameLoad attr NameLoad Assign SubscriptStore NameLoad Index CompareEq NameLoad Num Num Assert CompareGt Call AttributeLoad NameLoad attr NameLoad NameLoad Num decorator_list If CompareEq NameLoad Str body Expr Call NameLoad
N/A N/A numpy np N/A scipy.sparse sp sklearn metrics fastFM als get_test_problem N/A N/A task N/A regression N/A N/A X N/A N/A sp csc_matrix N/A N/A np array N/A N/A 6 1 N/A 2 3 N/A 3 0 N/A 6 1 N/A 4 5 dtype N/A np float64 N/A y N/A N/A np array N/A 298 266 29 298 848 dtype N/A np float64 N/A V N/A N/A np array N/A N/A 6 0 N/A 5 8 dtype N/A np float64 N/A w N/A N/A np array N/A 9 2 dtype N/A np float64 N/A w0 2 N/A N/A task classification N/A N/A y_labels N/A N/A np ones_like y N/A N/A y_labels N/A N/A y N/A N/A np median y -1 N/A y y_labels N/A N/A w0 w V y X N/A test_fm_classification_predict_proba N/A N/A N/A N/A N/A N/A w0 w V y X N/A get_test_problem task classification N/A fm N/A N/A als FMClassification n_iter 1000 init_stdev 0.1 l2_reg_w 0 l2_reg_V 0 rank 2 N/A N/A N/A fm fit X y N/A y_pred N/A N/A fm predict X N/A y_pred N/A N/A fm predict_proba X N/A N/A y N/A N/A y -1 0 N/A N/A N/A N/A metrics roc_auc_score y y_pred 0.95 N/A N/A N/A __name__ __main__ N/A N/A N/A test_fm_classification_predict_proba