Skip to content

Commit

Permalink
Update train.py
Browse files Browse the repository at this point in the history
  • Loading branch information
dec1costello authored Oct 13, 2024
1 parent 66508cc commit 27ad319
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions Streamlit/train.py
Original file line number Diff line number Diff line change
@@ -1,19 +1,19 @@
import numpy as np
import pandas as pd

import warnings
warnings.filterwarnings("ignore", category=UserWarning)
warnings.filterwarnings("ignore", category=DeprecationWarning)
warnings.filterwarnings("ignore", category=FutureWarning, module='category_encoders.base_contrast_encoder')
warnings.filterwarnings("ignore", category=pd.errors.SettingWithCopyWarning)

from sklearn.pipeline import Pipeline
from sklearn.impute import SimpleImputer
from category_encoders import HashingEncoder
from sklearn.compose import ColumnTransformer
from sklearn.preprocessing import StandardScaler
from sklearn.model_selection import train_test_split

import warnings
warnings.filterwarnings("ignore", category=UserWarning)
warnings.filterwarnings("ignore", category=DeprecationWarning)
warnings.filterwarnings("ignore", category=FutureWarning, module='category_encoders.base_contrast_encoder')
warnings.filterwarnings("ignore", category=pd.errors.SettingWithCopyWarning)

import optuna
from sklearn.datasets import make_regression
from sklearn.ensemble import GradientBoostingRegressor
Expand Down

0 comments on commit 27ad319

Please sign in to comment.