Hiperparâmetros de k-NN
A tabela a seguir lista os hiperparâmetros que você pode definir para o algoritmo k-nearest neighbors (k-NN) do Amazon SageMaker.
| Nome do parâmetro | Descrição |
|---|---|
feature_dim |
O número de atributos nos dados de entrada. Obrigatório Valores válidos: inteiro positivo. |
k |
O número de vizinhos mais próximos. Obrigatório Valores válidos: inteiro positivo |
predictor_type |
O tipo de inferência a ser usada nos rótulos de dados. Obrigatório Valores válidos: classificador para classificação ou regressor para regressão. |
sample_size |
O número de pontos de dados dos quais obter uma amostra no conjunto de dados de treinamento. Obrigatório Valores válidos: inteiro positivo |
dimension_reduction_target |
A dimensão de destino para a qual reduzir. Obrigatório quando você especifica o parâmetro Valores válidos: inteiro positivo maior que 0 e menor que |
dimension_reduction_type |
O tipo de método de redução da dimensão. Opcional Valores válidos: sign para projeção aleatória ou fjlt para a transformação rápida de Johnson-Lindenstrauss. Valor padrão: Nenhuma redução da dimensão |
faiss_index_ivf_nlists |
O número de centroides a serem construídos no índice quando Opcional Valores válidos: inteiro positivo Valor padrão: auto, que é resolvido como |
faiss_index_pq_m |
O número de subcomponentes vetoriais a serem construídos no índice quando A biblioteca FaceBook AI Similarity Search (FAISS) requer que o valor de Opcional Valores válidos: Um dos seguintes números inteiros positivos: 1, 2, 3, 4, 8, 12, 16, 20, 24, 28, 32, 40, 48, 56, 64, 96 |
index_metric |
A métrica para medir a distância entre os pontos ao encontrar os vizinhos mais próximos. Ao treinar com Opcional Valores válidos: L2 para distância euclidiana, INNER_PRODUCT para distância interna do produto, COSINE para similaridade de cosseno. Valor padrão: L2 |
index_type |
O tipo de índice. Opcional Valores válidos: faiss.Flat, faiss.IVFFlat, faiss.IVFPQ. Valores padrão: faiss.Flat |
mini_batch_size |
O número de observações por minilote para o iterador de dados. Opcional Valores válidos: inteiro positivo Valor padrão: 5000 |
