#!/usr/bin/env python
# coding: utf-8

# In[1]:


import altair
from IPython.display import display
from sklearn.neighbors import LocalOutlierFactor
from sklearn.datasets import load_iris, load_linnerud


# # Example A

# In[2]:


data = load_iris(as_frame=True).data
data["Outlier"] = LocalOutlierFactor(50).fit_predict(data) == -1
display(data[:3])
altair.Chart(data).mark_point().encode(
    x="sepal length (cm)",
    y="petal length (cm)",
    color="Outlier"
)


# # Example B

# In[3]:


data = load_linnerud(as_frame=True).data
data["Outlier"] = LocalOutlierFactor(10).fit_predict(data) == -1
display(data[:3])
altair.Chart(data).mark_point().encode(
    x="Chins",
    y="Jumps",
    color="Outlier"
)