Add files via upload

Author: PyCaret

Dockerfile

@@ -10,4 +10,4 @@ RUN pip install --trusted-host pypi.python.org -r requirements.txt
 
 ENTRYPOINT ["python"]
 
-CMD ["/app/app.py"]
+CMD ["/app/app.py"]

action.yml

@@ -5,7 +5,7 @@ inputs:
   myInput:
     description: "Classification"
     required: true
-    default: "zeroinputs"
+    default: "target-variable"
 outputs:
   myOutput:
     description: "Output from the action"

app.py

@@ -5,11 +5,13 @@
 data = pd.read_csv(dataset_path)
 data.head()
 
+target = ast.literal_eval(os.environ["INPUT_MYINPUT"])
+
 from pycaret.classification import *
-clf1 = setup(data, target = 'Purchase', session_id=123, silent=True, html=False, log_experiment=True, experiment_name='juice1')
+clf1 = setup(data, target =  target, session_id=123, silent=True, html=False, log_experiment=True, experiment_name='exp1')
 
 c = compare_models()
 
 save_model(c, 'model')
 
-logs_juice1 = get_logs(save=True)
+logs_exp1 = get_logs(save=True)

requirements.txt

@@ -1 +1 @@
-pycaret-nightly==0.20
+pycaret-nightly==0.24