Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import pandas as pd
- from sklearn.ensemble import IsolationForest
- df = pd.read_csv('/datasets/sales.csv')
- data = df[['Sales', 'Profit']]
- sales = df['Sales'].values.reshape(-1, 1)
- profit = df['Profit'].values.reshape(-1, 1)
- isolation_forest = IsolationForest(n_estimators=100)
- isolation_forest.fit(data)
- #estimator = isolation_forest.predict(data)
- estimator = isolation_forest.fit_predict(data)
- outliers = (estimator == -1).tolist()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement