Care algoritm de învățare automată pentru clasificare ai încredere?

Pentru sarcinile de clasificare, se alege adesea algoritmul de ** regresie logistică** sau arbore de decizie datorită simplității și interpretabilității lor.

De exemplu, regresia logistică este potrivită pentru clasificarea binară și se antrenează rapid pe seturi de date mici și medii. Modelează probabilitatea ca un obiect să aparțină unei clase printr-o funcție sigmoidală.

Exemplu în Python cu sklearn:

from sklearn.linear_model import LogisticRegression
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split

# Încărcăm datele
X, y = load_iris(return_X_y=True)

# Pentru simplificare, luăm doar două clase
X = X[y != 2]
y = y[y != 2]

# Împărțim în seturi de antrenament și test
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)

# Antrenăm modelul
model = LogisticRegression()
model.fit(X_train, y_train)

# Evaluăm acuratețea
print("Acuratețe:", model.score(X_test, y_test))

Acest algoritm este ușor de implementat și adesea servește ca un punct de plecare bun pentru sarcinile de clasificare.