Sobes.tech
Middle — Senior

Vali masinõppe algoritm klassifikatsioonitöö jaoks, millele sa usaldad?

sobes.tech AI

Vastus AI-lt

Klassifikatsiooni ülesannete jaoks valitakse sageli logistiline regressioon või otsustuspuu algoritm, kuna need on lihtsad ja tõlgendatavad.

Näiteks sobib logistiline regressioon hästi binaarseteks klassifikatsioonideks ning õppib kiiresti väikeste ja keskmise suurusega andmekogumite peal. See modelleerib objekti klassi kuuluvust sigmoidfunktsiooni abil.

Näide Pythonis sklearn kasutades:

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

# Andmete laadimine
X, y = load_iris(return_X_y=True)

# Lihtsuse huvides võtame ainult kaks klassi
X = X[y != 2]
y = y[y != 2]

# Jagamine treening- ja testkomplektideks
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)

# Mudeli treenimine
model = LogisticRegression()
model.fit(X_train, y_train)

# Täpsuse hindamine
print("Täpsus:", model.score(X_test, y_test))

See algoritm on lihtne rakendada ning sageli on see hea lähtepunkt klassifikatsiooni ülesannete jaoks.