{"cells":[{"metadata":{"trusted":true},"cell_type":"code","source":"import riiideducation\nimport pickle\nimport pandas as pd\nimport numpy as np","execution_count":null,"outputs":[]},{"metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","trusted":true},"cell_type":"code","source":"with open('../input/multipledt1/decision_tree.pkl', 'rb') as f:\n    dt_list, ques, user = pickle.load(f)","execution_count":null,"outputs":[]},{"metadata":{"_uuid":"d629ff2d2480ee46fbb7e2d37f6b5fab8052498a","_cell_guid":"79c7e3d0-c299-4dcb-8224-4455121ee9b0","trusted":true},"cell_type":"code","source":"import riiideducation\n\n#Create the env\nenv = riiideducation.make_env()\n\n#Create the iterator\niter_test = env.iter_test()\n\n#Iter and predict\nfor (test_df, sample_prediction_df) in iter_test:\n    test_df_q = pd.merge(test_df, ques, how='left',on='content_id')\n    test_df_q = pd.merge(test_df_q, user,  how='left', on='user_id')\n    test_df_q['q_difficulty'] = test_df_q['q_+_response']/ test_df_q['total_responses_to_q']\n    test_df_q['user_stats'] = test_df_q['u_+_response']/ test_df_q['total_user_responses']\n    res = np.zeros([len(test_df_q),2])\n    for dt in range(len(dt_list)):\n        dt_now = dt_list[dt]\n        idx = dt_now.predict(test_df_q[['total_responses_to_q', 'q_difficulty', 'total_user_responses', 'user_stats']].fillna(0))\n        for j in range(len(test_df_q)):\n            res[j,idx[j]] += 1\n    test_df['answered_correctly'] = np.argmax(res, axis=1)\n    env.predict(test_df.loc[test_df['content_type_id'] == 0, ['row_id', 'answered_correctly']])","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"","execution_count":null,"outputs":[]}],"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"pygments_lexer":"ipython3","nbconvert_exporter":"python","version":"3.6.4","file_extension":".py","codemirror_mode":{"name":"ipython","version":3},"name":"python","mimetype":"text/x-python"}},"nbformat":4,"nbformat_minor":4}