python

pythin.py

import sys
import os
import numpy as np
import pandas as pd
from tqdm.auto import tqdm
import matplotlib.pyplot as plt

from scipy.spatial.distance import pdist, squareform
from itertools import chain

from catboost import CatBoostRegressor
from lightgbm import LGBMRegressor

from scipy.signal import find_peaks
from scipy.stats import mode

os.environ["CUDA_VISIBLE_DEVICES"] = "0"


train_df = pd.read_csv("../data/train.csv", index_col="id")
train_df['epoch'] = pd.to_datetime(train_df['epoch'])

test_df = pd.read_csv("../data/test.csv", index_col="id")

test_sats = test_df['sat_id'].unique()
for sat_id in test_sats:
    train_subset = train_df[train_df['sat_id'] == sat_id]
    assert train_subset['epoch'].is_monotonic

train_df.head()