Classification, Catboost, Fastapi, Docker, Streamlit.
StratifiedShuffleSplit(n_splits=1, test_size=0.2, random_state=64) train_index, test_index = next(strat_split.split(df, df["Churn"])) # Create train and test sets strat_train_set = df.loc[train_index] strat_test_set = df.loc[test_index] X_train = strat_train_set.drop("Churn", axis=1) y_train =