Data, code and results of 'A Search for a More Robust AGN Classifier Using Machine Learning'

Data, code, and resulting classifiers for selecting QSOs nsa_dataset.fits - contains full set of collected data; used for training and testingnsa_reduced_data.fits - contains collected data excluding ambiguous cases (objects in the BPT Composite region with no further AGN evidence); used for full testing performance on a larger population gradient_boosted_trees.py - python module containing functions used to train, predict, and evaluate gradient boosted trees classifiersneural_net.py - python module containing functions used to train, predict, and evaluate multi-layer perceptrons/neural nets xgb_best.json - XGBoost classifier object trained with the "best" feature setxgb_phot.json - XGBoost classifier object trained with the "best photometric" feature setxgb_spec.json - XGBoost classifier object trained with the "best spectroscopic" feature set mlp_best.pkl - scikit-learn MLPClassifier object trained with the "best" feature setmlp_phot.pkl - scikit-learn MLPClassifier object trained with the "best photometric" feature setmlp_spec.pkl - scikit-learn MLPClassifier object trained with the "best spectroscopic" feature set ### DATA FORMAT ###Name Type Unit Description---------------------------------------------------Name string Name assigned to objectSource string Source of objectF_mag float mag SDSS F-band magnitudeN_mag float mag SDSS N-band magnitudeu_mag float mag SDSS u-band magnitudeg_mag float mag SDSS g-band magnituder_mag float mag SDSS r-band magnitudei_mag float mag SDSS i-band magnitudez_mag float mag SDSS z-band magnitudeW1_mag float mag WISE W1-band magnitudeW2_mag float mag WISE W2-band magnitudeW3_mag float mag WISE W3-band magnitudeW4_mag float mag WISE W4-band magnitudeF-N float mag Difference between F and N magnitudesN-u float mag Difference between N and u magnitudesu-g float mag Difference between u and g magnitudesg-r float mag Difference between g and r magnitudesr-i float mag Difference between r and i magnitudesi-z float mag Difference between i and z magnitudesz-W1 float mag Difference between z and W1 magnitudesW1-W2 float mag Difference between W1 and W2 magnitudesW2-W3 float mag Difference between W2 and W3 magnitudesW3-W4 float mag Difference between W3 and W4 magnitudesF-u float mag Difference between F and u magnitudesN-g float mag Difference between N and g magnitudesu-r float mag Difference between u and r magnitudesg-i float mag Difference between g and i magnitudesr-z float mag Difference between r and z magnitudesi-W1 float mag Difference between i and W1 magnitudesz-W2 float mag Difference between z and W2 magnitudesW1-W3 float mag Difference between W1 and W3 magnitudesW2-W4 float mag Difference between W2 and W4 magnitudesF-g float mag Difference between F and g magnitudesN-r float mag Difference between N and r magnitudesu-i float mag Difference between u and i magnitudesg-z float mag Difference between g and z magnitudesr-W1 float mag Difference between r and W1 magnitudesi-W2 float mag Difference between i and W2 magnitudesz-W3 float mag Difference between z and W3 magnitudesW1-W4 float mag Difference between W1 and W4 magnitudesF-r float mag Difference between F and r magnitudesN-i float mag Difference between N and i magnitudesu-z float mag Difference between u and z magnitudesg-W1 float mag Difference between g and W1 magnitudesr-W2 float mag Difference between r and W2 magnitudesi-W3 float mag Difference between i and W3 magnitudesz-W4 float mag Difference between z and W4 magnitudesF-i float mag Difference between F and i magnitudesN-z float mag Difference between N and z magnitudesu-W1 float mag Difference between u and W1 magnitudesg-W2 float mag Difference between g and W2 magnitudesr-W3 float mag Difference between r and W3 magnitudesi-W4 float mag Difference between i and W4 magnitudesF-z float mag Difference between F and z magnitudesN-W1 float mag Difference between N and W1 magnitudesu-W2 float mag Difference between u and W2 magnitudesg-W3 float mag Difference between g and W3 magnitudesr-W4 float mag Difference between r and W4 magnitudesF-W1 float mag Difference between F and W1 magnitudesN-W2 float mag Difference between N and W2 magnitudesu-W3 float mag Difference between u and W3 magnitudesg-W4 float mag Difference between g and W4 magnitudesF-W2 float mag Difference between F and W2 magnitudesN-W3 float mag Difference between N and W3 magnitudesu-W4 float mag Difference between u and W4 magnitudesF-W3 float mag Difference between F and W3 magnitudesN-W4 float mag Difference between N and W4 magnitudesF-W4 float mag Difference between F and W4 magnitudesHbeta_n float 10-17 erg/s/cm2 H beta fluxOIII5007 float 10-17 erg/s/cm2 [OIII] 5007 fluxOI6300 float 10-17 erg/s/cm2 [OI] 6300 fluxHalpha_n float 10-17 erg/s/cm2 H alpha fluxNII6583 float 10-17 erg/s/cm2 [NII] 6584 fluxSII6716 float 10-17 erg/s/cm2 [SII] 6716 fluxSII6731 float 10-17 erg/s/cm2 [SII] 6731 fluxEW_Hbeta_n float H beta equivalent widthEW_OIII5007 float [OIII] 5007 equivalent widthEW_OI6300 float [OI] 6300 equivalent widthEW_Halpha_n float H alpha equivalent widthEW_NII6583 float [NII] 6584 equivalent widthEW_SII6716 float [SII] 6716 equivalent widthEW_SII6731 float [SII] 6731 equivalent widthN2/Ha float Logarithm of the ratio of [NII] 6584 and H alpha fluxO3/Hb float Logarithm of the ratio of [OIII] 5007 and H beta fluxHb/O1 float Logarithm of the ratio of H beta and [OI] 6300 fluxHb/Ha float Logarithm of the ratio of H beta and H alpha fluxHb/N2 float Logarithm of the ratio of H beta and [NII] 6584 fluxHb/S2_6716 float Logarithm of the ratio of H beta and [SII] 6716 fluxHb/S2_6731 float Logarithm of the ratio of H beta and [SII] 6731 fluxO3/O1 float Logarithm of the ratio of [OIII] 5007 and H alpha fluxO3/Ha float Logarithm of the ratio of [OIII] 5007 and H alpha fluxO3/N2 float Logarithm of the ratio of [OIII] 5007 and [NII] 6584 fluxO3/S2_6716 float Logarithm of the ratio of [OIII] 5007 and [SII] 6716 fluxO3/S2_6731 float Logarithm of the ratio of [OIII] 5007 and [SII] 6731 fluxO1/Ha float Logarithm of the ratio of [OI] 6300 and H alpha fluxO1/N2 float Logarithm of the ratio of [OI] 6300 and [NII] 6584 fluxO1/S2_6716 float Logarithm of the ratio of [OI] 6300 and [SII] 6716 fluxO1/S2_6731 float Logarithm of the ratio of [OI] 6300 and [SII] 6731 fluxHa/S2_6716 float Logarithm of the ratio of H alpha and [SII] 6716 fluxHa/S2_6731 float Logarithm of the ratio of H alpha and [SII] 6731 fluxN2/S2_6716 float Logarithm of the ratio of [NII] 6584 and [SII] 6716 fluxN2/S2_6731 float Logarithm of the ratio of [NII] 6584 and [SII] 6731 fluxS2_6716/S2_6731 float Logarithm of the ratio of [SII] 6716 and [SII] 6731 fluxz_spec float Spectroscopic redshiftv_dis float km/s Velocity dispersonDn4000 float Dn4000 parameterlogMstar float log(M_sun) Logarithm of galaxy mass (in solar masses)BPT string BPT diagnostic classificationQSO int QSO classification (1 = QSO, 0 = inactive) ### DATA SOURCES ###https://ui.adsabs.harvard.edu/abs/2024ApJ...971...68Whttps://ui.adsabs.harvard.edu/abs/2013ApJ...775..116Rhttps://ui.adsabs.harvard.edu/abs/2014AJ....148..136Mhttps://ui.adsabs.harvard.edu/abs/2015MNRAS.454.3722Shttps://ui.adsabs.harvard.edu/abs/2018ApJ...863....1Chttps://ui.adsabs.harvard.edu/abs/2018ApJ...868..152Bhttps://ui.adsabs.harvard.edu/abs/2020MNRAS.492.2268Bhttps://ui.adsabs.harvard.edu/abs/2020ApJ...896...10Bhttps://ui.adsabs.harvard.edu/abs/2022ApJ...933...37Whttps://ui.adsabs.harvard.edu/abs/2022ApJ...937....7Shttps://ui.adsabs.harvard.edu/abs/2022MNRAS.516.2736Bhttps://ui.adsabs.harvard.edu/abs/2020ApJS..250....8Lhttp://sdss.physics.nyu.edu/mblanton/v0/nsa_v0_1_2.fitshttps://data.sdss.org/sas/dr17/sdss/atlas/v1/nsa_v1_0_1.fitshttps://ui.adsabs.harvard.edu/abs/2010AJ....140.1868W