Skip to content

Commit 7778178

Browse files
Update __init__.py
1 parent d74db3b commit 7778178

File tree

1 file changed

+40
-2
lines changed

1 file changed

+40
-2
lines changed

DimRed/__init__.py

Lines changed: 40 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,51 @@
1+
from sklearn.pipeline import make_pipeline
12
import pandas as pd
23
import matplotlib.pyplot as plt
4+
from sklearn.metrics import classification_report
5+
from sklearn.preprocessing import LabelEncoder
36
import numpy as np
7+
from pprint import pprint
8+
import json
9+
import wandb
410
from sklearn.model_selection import ParameterGrid
511
# TODO make_pipeline is excessive
12+
from sklearn.model_selection import train_test_split
613
from sklearn.preprocessing import StandardScaler
14+
import xgboost as xgb
15+
import lightgbm as lgb
16+
from sklearn.linear_model import LogisticRegression
17+
from sklearn.svm import SVC
18+
from sklearn.tree import DecisionTreeClassifier
19+
from sklearn.ensemble import RandomForestClassifier
20+
from sklearn.neighbors import KNeighborsClassifier
21+
import threading
22+
import cupy as cp
23+
import warnings
24+
from dotenv import load_dotenv
25+
from tqdm import tqdm
726
from sklearn.pipeline import Pipeline, make_pipeline
827
from typing import *
9-
from sklearn.pipeline import make_pipeline
28+
import seaborn as sns
29+
from sklearn.model_selection import GridSearchCV
30+
import math
31+
import os
32+
import random
33+
from sklearn.decomposition import PCA, IncrementalPCA
34+
from sklearn import datasets
35+
from wandb.lightgbm import wandb_callback, log_summary
36+
import logging
37+
from wandb.xgboost import WandbCallback
38+
load_dotenv()
39+
PROJECT_NAME = os.getenv("PROJECT_NAME")
40+
logging.getLogger('lightgbm').setLevel(logging.WARNING)
41+
logging.getLogger("wandb").setLevel(logging.ERROR)
42+
os.environ["WANDB_SILENT"] = "true"
43+
warnings.filterwarnings("ignore")
44+
RANDOM_STATE = 42
45+
random.seed(RANDOM_STATE)
46+
np.random.seed(RANDOM_STATE)
47+
from DimRed.config import *
1048
from DimRed.helper_functions import *
49+
from DimRed.analysis import *
1150
from DimRed.variation import *
1251
from DimRed.evaluation import *
13-
from DimRed.analysis import *

0 commit comments

Comments
 (0)