Skip to content

Commit

Permalink
Adding automl
Browse files Browse the repository at this point in the history
  • Loading branch information
James Bristow committed Mar 1, 2024
1 parent a50e569 commit fac5db6
Show file tree
Hide file tree
Showing 2 changed files with 42 additions and 0 deletions.
Binary file added AUC.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
42 changes: 42 additions & 0 deletions auto_ml_test.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,42 @@
import os
os.environ['WHYLOGS_NO_ANALYTICS']='True'

from deepchecks.tabular.suites import full_suite
from evidently.metric_preset import (
DataDriftPreset, DataQualityPreset, TargetDriftPreset, RegressionPreset, ClassificationPreset
)

from evidently.report import Report
from evidently.test_preset import (
DataStabilityTestPreset, DataQualityTestPreset,
BinaryClassificationTestPreset, BinaryClassificationTopKTestPreset,
DataDriftTestPreset, MulticlassClassificationTestPreset,
RegressionTestPreset
)

from evidently.test_suite import TestSuite
from sklearn.datasets import load_iris
from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import train_test_split
import whylogs as why
from whylogs.viz import NotebookProfileVisualizer

from ydata_profiling import ProfileReport

from pycaret.classification import *

def main():
# load dataset
from pycaret.datasets import get_data
diabetes = get_data('diabetes')

clf1 = setup(data = diabetes, target = 'Class variable')

# creating a model
lr = create_model('lr')

plot_model(lr, plot = 'auc', save = True)


if __name__ == "__main__":
main()

0 comments on commit fac5db6

Please sign in to comment.