{"cells":[{"metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","trusted":true},"cell_type":"code","source":"import librosa\nimport numpy as np\nimport pandas as pd\nfrom sklearn import *\nfrom PIL import Image\nfrom numba import njit\nimport soundfile as sf\nfrom scipy.io import wavfile\nimport scipy, cv2, imagehash\nimport matplotlib.pyplot as plt\nfrom multiprocessing import Pool, cpu_count\n%matplotlib inline\n\nlabel = {'FAKE':1, 'REAL':0}\npath = '/kaggle/input/deepfake-detection-challenge/train_sample_videos/'\ntrain = pd.read_json(path + 'metadata.json').T.reset_index()\ntrain.columns = ['filename', 'label', 'split', 'original']\ntrain['path'] = path + train['filename']\ntrain['label'] = train['label'].map(label)\n\npath = '/kaggle/input/deepfake-detection-challenge/'\ntest = pd.read_csv(path + 'sample_submission.csv')\ntest['path'] = path + 'test_videos/' + test['filename']\nprint(train.shape, test.shape)","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"train['file_meta'] = train.path.map(lambda x: open(x,'rb').read(1000))\ntest['file_meta'] = test.path.map(lambda x: open(x,'rb').read(1000))","execution_count":null,"outputs":[]},{"metadata":{"trusted":true},"cell_type":"code","source":"import xgboost as xgb\nimport joblib, pickle\n\nft_cv2 = pickle.load(open('/kaggle/input/deepfake/cv.pickle','rb'))\nmodel2 = joblib.load('/kaggle/input/deepfake/model.xgb')\n\ntest['label'] = model2.predict(xgb.DMatrix(ft_cv2.transform(test.file_meta)), ntree_limit=model2.best_ntree_limit)\ntest[['filename', 'label']].to_csv('submission.csv', index=False)","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":1}