Add files via upload

Author: PyCaret

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.7-slim
+
+WORKDIR /app
+
+ADD . /app
+
+RUN pip install --trusted-host pypi.python.org -r requirements.txt
+
+ENTRYPOINT ["python"]
+
+CMD ["/app/app.py"]

action.yml

@@ -0,0 +1,17 @@
+name: "PyCaret AutoML Git Action"
+description: "pycaret.classification"
+author: "Moez Ali"
+inputs:
+  myInput:
+    description: "Classification"
+    required: true
+    default: "zeroinputs"
+outputs:
+  myOutput:
+    description: "Output from the action"
+runs:
+  using: "docker"
+  image: "Dockerfile"
+branding:
+  icon: 'box'
+  color: 'blue'

app.py

@@ -0,0 +1,13 @@
+import os, ast
+import pandas
+
+dataset_path = "https://raw.githubusercontent.com/" + os.environ["GITHUB_REPOSITORY"] +"/master/dataset.csv"
+data = pd.read_csv(dataset_path)
+data.head()
+
+from pycaret.classification import *
+clf1 = setup(data, target = 'Purchase', session_id=123, silent=True, html=False)
+
+c = compare_models()
+
+save_model(c, 'Trained Model')

requirements.txt

@@ -0,0 +1 @@
+pycaret-nightly