{"cells":[{"metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","trusted":true},"cell_type":"code","source":"try:\n    import corner\nexcept:\n    !pip install corner\n    import corner\n    \nimport numpy as np\nimport pandas as pd","execution_count":null,"outputs":[]},{"metadata":{},"cell_type":"markdown","source":"### 1. training data corner plots"},{"metadata":{"trusted":true},"cell_type":"code","source":"import matplotlib.pyplot as plt \nimport numpy as np\nimport pandas as pd\n\ndata = pd.read_csv('../input/PLAsTiCC-2018/training_set_metadata.csv')\ndata_z = data[['hostgal_specz','hostgal_photoz']]\n# log_data_z = pd.concat([np.log(data_z['hostgal_specz']),np.log(data_z['hostgal_photoz'])], axis = 1)\nfig = corner.corner(data_z,range=[(0,2.4),(0,2.4)], show_titles=True, title_kwargs={\"fontsize\":10},label_kwargs = {'fontsize':14})\nfor ax in fig.get_axes():   \n      ax.tick_params(axis='both', labelsize=14)\n\ndel data_z","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"data_photoz = data[['hostgal_photoz','hostgal_photoz_err']]\nfigure = corner.corner(data_photoz, show_titles=True, title_kwargs={\"fontsize\": 12})\ndel data_photoz\n","execution_count":null,"outputs":[]},{"metadata":{},"cell_type":"markdown","source":"### 2. training data AGN corner plots"},{"metadata":{"trusted":true},"cell_type":"code","source":"data = pd.read_csv('../input/PLAsTiCC-2018/training_set_metadata.csv')\nagn_data = data[data.target==88]\ndata_z = agn_data[['hostgal_specz','hostgal_photoz']]\nlog_data_z = pd.concat([np.log10(data_z['hostgal_specz']),np.log10(data_z['hostgal_photoz'])], axis = 1)\nfig = corner.corner(data_z, labels = ['log(specz)','log(photoz)'],show_titles=True, title_kwargs={\"fontsize\":10},label_kwargs = {'fontsize':14})\nfor ax in fig.get_axes():   \n      ax.tick_params(axis='both', labelsize=14)\ndel data_z","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"data_photoz = agn_data[['hostgal_photoz','hostgal_photoz_err']]\nfigure = corner.corner(data_photoz, show_titles=True, title_kwargs={\"fontsize\": 12})\ndel data_photoz","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"lc_data = pd.read_csv('../input/PLAsTiCC-2018/training_set.csv')\nlc_data = lc_data[lc_data.object_id.isin(agn_data.object_id)]\nu_data = lc_data[lc_data.passband == 0][['flux','flux_err']]\nfigure = corner.corner(u_data, range = [(-300,300),(0,100)],show_titles=True, title_kwargs={\"fontsize\": 12})","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"g_data = lc_data[lc_data.passband == 1][['flux','flux_err']]\nfigure = corner.corner(g_data, range = [(-300,300),(0,40)], show_titles=True, title_kwargs={\"fontsize\": 12})","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"r_data = lc_data[lc_data.passband == 2][['flux','flux_err']]\nfigure = corner.corner(r_data, range = [(-300,300),(0,40)], show_titles=True, title_kwargs={\"fontsize\": 12})","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"i_data = lc_data[lc_data.passband == 3][['flux','flux_err']]\nfigure = corner.corner(i_data, range = [(-300,300),(0,40)], show_titles=True, title_kwargs={\"fontsize\": 12})","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"z_data = lc_data[lc_data.passband == 4][['flux','flux_err']]\nfigure = corner.corner(z_data, range = [(-300,300),(0,60)], show_titles=True, title_kwargs={\"fontsize\": 12})","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"y_data = lc_data[lc_data.passband == 5][['flux','flux_err']]\nfigure = corner.corner(y_data, range = [(-300,300),(0,100)], show_titles=True, title_kwargs={\"fontsize\": 12})","execution_count":null,"outputs":[]},{"metadata":{},"cell_type":"markdown","source":"### 3. training data six bands corner plots"},{"metadata":{"trusted":true},"cell_type":"code","source":"# data_convert = pd.read_csv('../input/plasticc-converted-datasets/training_set_converted.csv')\n\n# figure = corner.corner(data_convert[['u','g','r','i','z','y']], show_titles=True, title_kwargs={\"fontsize\": 12})","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}