Middle — Senior
Pasirinkite mašininio mokymosi algoritmą klasifikacijos užduočiai, kuriuo pasitikite?
sobes.tech AI
Atsakymas iš AI
Klasifikacijos užduotims dažnai pasirenkamas logistinė regresija arba sprendimų medis dėl jų paprastumo ir interpretabilumo.
Pavyzdžiui, logistinė regresija yra tinkama binarinei klasifikacijai ir greitai mokosi iš mažų ir vidutinių duomenų rinkinių. Ji modeliuoja objekto priklausymą klasei per sigmoidinę funkciją.
Pavyzdys Python kalba naudojant sklearn:
from sklearn.linear_model import LogisticRegression
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
# Duomenų įkėlimas
X, y = load_iris(return_X_y=True)
# Paprastumo dėlei paimkime tik dvi klases
X = X[y != 2]
y = y[y != 2]
# Duomenų padalijimas į mokymo ir testavimo rinkinius
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)
# Modelio mokymas
model = LogisticRegression()
model.fit(X_train, y_train)
# Tikslumo įvertinimas
print("Tikslumas:", model.score(X_test, y_test))
Šis algoritmas yra lengvai įgyvendinamas ir dažnai tarnauja kaip geras pradinis taškas klasifikavimo užduotims.