File size: 458 Bytes
06b052d
 
 
 
 
 
 
 
 
 
 
 
 
0f6f971
4c3160a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#One-Class Support Vector Machine

from sklearn import svm
import numpy as np
from sklearn.discriminant_analysis import StandardScaler

def apply_oc_svm(all_data: np.ndarray) -> np.ndarray:
    # Normalize
    scaler = StandardScaler()
    X_scaled = scaler.fit_transform(all_data)

    # Initialize One-Class SVM
    oc_svm = svm.OneClassSVM(kernel='rbf', gamma='auto', nu=0.2)
    print(oc_svm.fit_predict(X_scaled))
    return oc_svm.fit_predict(X_scaled)