LDAHyperparamètres - Amazon SageMaker AI

Les traductions sont fournies par des outils de traduction automatique. En cas de conflit entre le contenu d'une traduction et celui de la version originale en anglais, la version anglaise prévaudra.

LDAHyperparamètres

Dans la demande CreateTrainingJob, vous spécifiez l'algorithme d'entraînement. Vous pouvez également spécifier des hyperparamètres spécifiques à l'algorithme sous forme de cartes. string-to-string Le tableau suivant répertorie les hyperparamètres de l'algorithme LDA d'entraînement fourni par Amazon SageMaker AI. Pour de plus amples informations, veuillez consulter Fonctionnement de LDA.

Nom du paramètre Description
num_topics

Le nombre de sujets LDA à rechercher dans les données.

Obligatoire

Valeurs valides : nombre entier positif

feature_dim

Taille du vocabulaire du corpus de documents d'entrée.

Obligatoire

Valeurs valides : nombre entier positif

mini_batch_size

Nombre total de documents dans le corpus de documents d'entrée.

Obligatoire

Valeurs valides : nombre entier positif

alpha0

Supposition initiale pour le paramètre de concentration : somme des éléments de l'antécédent Dirichlet. Les petites valeurs sont plus susceptibles de générer des mélanges de rubriques dispersés et les valeurs élevées (supérieures à 1,0), des mélanges uniformes.

Facultatif

Valeurs valides : valeur flottante positive

Valeur par défaut : 1.0

max_restarts

Nombre de redémarrages à effectuer pendant la phase de décomposition spectrale des moindres carrés alternatifs (ALS) de l'algorithme. Peut être utilisé pour trouver des minima locaux de meilleure qualité sous condition de calculs supplémentaires, mais ne doit pas être ajusté en général.

Facultatif

Valeurs valides : nombre entier positif

Valeur par défaut : 10

max_iterations

Le nombre maximal d'itérations à effectuer pendant la ALS phase de l'algorithme. Peut être utilisé pour trouver des minima de meilleure qualité sous condition de calculs supplémentaires, mais ne doit pas être ajusté en général.

Facultatif

Valeurs valides : nombre entier positif

Valeur par défaut : 1000

tol

Tolérance d'erreur cible pour la ALS phase de l'algorithme. Peut être utilisé pour trouver des minima de meilleure qualité sous condition de calculs supplémentaires, mais ne doit pas être ajusté en général.

Facultatif

Valeurs valides : valeur flottante positive

Valeur par défaut : 1e-8