{ "cells": [ { "cell_type": "code", "execution_count": 1, "metadata": {}, "outputs": [], "source": [ "import pyarrow.parquet as pq\n", "import numpy as np\n", "import pandas as pd\n", "from tensorflow.keras.layers import *\n", "from tensorflow.keras.layers import Concatenate\n", "from tensorflow.keras.models import Sequential, Model\n", "from tensorflow.keras.optimizers import Adam\n", "from tensorflow.keras.utils import to_categorical\n", "from tensorflow.keras.metrics import AUC\n", "import tensorflow as tf\n", "import warnings\n", "warnings.filterwarnings('ignore')\n" ] }, { "cell_type": "code", "execution_count": 2, "metadata": {}, "outputs": [], "source": [ "file = 'QCDToGGQQ_IMGjet_RH1all_jet0_run0_n36272.test.snappy.parquet'" ] }, { "cell_type": "code", "execution_count": 3, "metadata": {}, "outputs": [], "source": [ "data = pq.read_table(file)" ] }, { "cell_type": "code", "execution_count": 4, "metadata": {}, "outputs": [], "source": [ "data = data.to_pandas()" ] }, { "cell_type": "code", "execution_count": 5, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
| \n", " | X_jets | \n", "pt | \n", "m0 | \n", "y | \n", "
|---|---|---|---|---|
| 0 | \n", "[[[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0... | \n", "112.411095 | \n", "21.098248 | \n", "0.0 | \n", "
| 1 | \n", "[[[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0... | \n", "95.220406 | \n", "14.030600 | \n", "1.0 | \n", "
| 2 | \n", "[[[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0... | \n", "97.007317 | \n", "17.728968 | \n", "1.0 | \n", "
| 3 | \n", "[[[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0... | \n", "82.490311 | \n", "14.702741 | \n", "0.0 | \n", "
| 4 | \n", "[[[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.328483... | \n", "102.539238 | \n", "19.456257 | \n", "0.0 | \n", "