Categories
kaggle

Starter Notebook Google – Fast or Slow? Predict AI Model Runtime

このコンペでは、GNNを用いた手法で、Starter Notebookが提供されています。ここでは、そのStarter Notebookの内容を解説していきます。
元のページはここにあります。

ライブラリのインストール

!pip install tensorflow_gnn --pre
!pip install tensorflow_ranking

ライブラリのインポート

import os
import numpy as np
import pandas as pd
import tensorflow as tf
import tensorflow_gnn as tfgnn
import tensorflow_ranking as tfr

import tpugraphsv1_layout_data_py as layout_data
import tpugraphsv1_tile_data_py as tile_data
import tpugraphsv1_implicit_py as implicit

参考

https://www.kaggle.com/code/gusthema/starter-notebook-fast-or-slow-with-tensorflow-gnn