Skip to content

Commit

Permalink
Merge remote-tracking branch 'origin/master'
Browse files Browse the repository at this point in the history
  • Loading branch information
sronilsson committed Aug 7, 2024
2 parents ad75a0a + 486b895 commit f1aee91
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions simba/mixins/train_model_mixin.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,25 +16,26 @@
from copy import deepcopy
from datetime import datetime
from itertools import repeat
from subprocess import call
from json import loads
from subprocess import call

import numpy as np
import pandas as pd
import seaborn as sns
import shap
from imblearn.combine import SMOTEENN
from imblearn.over_sampling import SMOTE
from numba import njit, typed, types
from sklearn.ensemble import RandomForestClassifier
from sklearn.feature_selection import VarianceThreshold
from sklearn.inspection import partial_dependence, permutation_importance
from sklearn.metrics import precision_recall_curve, classification_report
from sklearn.metrics import classification_report, precision_recall_curve
from sklearn.model_selection import ShuffleSplit, learning_curve
from sklearn.preprocessing import (MinMaxScaler, QuantileTransformer, StandardScaler)
from sklearn.preprocessing import (MinMaxScaler, QuantileTransformer,
StandardScaler)
from sklearn.tree import export_graphviz
from sklearn.utils import parallel_backend
from tabulate import tabulate
import seaborn as sns

try:
from dtreeviz.trees import dtreeviz, tree
Expand All @@ -52,6 +53,7 @@
except:
from typing_extensions import Literal

from simba.mixins.plotting_mixin import PlottingMixin
from simba.plotting.shap_agg_stats_visualizer import \
ShapAggregateStatisticsVisualizer
from simba.ui.tkinter_functions import TwoOptionQuestionPopUp
Expand All @@ -77,7 +79,6 @@
MultiProcessingFailedWarning,
NoModuleWarning, NotEnoughDataWarning,
SamplingWarning, ShapWarning)
from simba.mixins.plotting_mixin import PlottingMixin

plt.switch_backend("agg")

Expand Down

0 comments on commit f1aee91

Please sign in to comment.