Catat tanggalnya! Google I / O mengembalikan 18-20 Mei Daftar sekarang
Halaman ini diterjemahkan oleh Cloud Translation API.
Switch to English

Rekomendasi

Rekomendasi hasil personalisasi banyak digunakan untuk berbagai kasus penggunaan di perangkat seluler, seperti pengambilan konten media, saran produk belanja, dan rekomendasi aplikasi berikutnya. Jika Anda tertarik untuk memberikan rekomendasi yang dipersonalisasi dalam aplikasi Anda sambil menghormati privasi pengguna, kami sarankan untuk menjelajahi contoh dan toolkit berikut.

Memulai

Kami menyediakan aplikasi sampel TensorFlow Lite yang menunjukkan cara merekomendasikan item yang relevan kepada pengguna di Android.

Contoh Android

Jika Anda menggunakan platform selain Android, atau Anda sudah terbiasa dengan TensorFlow Lite API, Anda dapat mendownload model rekomendasi awal kami.

Unduh model pemula

Kami juga menyediakan skrip pelatihan di Github untuk melatih model Anda sendiri.

Kode pelatihan

Pahami arsitektur model

Kami memanfaatkan arsitektur model encoder ganda, dengan encoder konteks untuk mengenkode riwayat pengguna berurutan dan label-encoder untuk mengenkode kandidat rekomendasi yang diprediksi. Kesamaan antara konteks dan pengkodean label digunakan untuk mewakili kemungkinan bahwa kandidat yang diprediksi memenuhi kebutuhan pengguna.

Tiga teknik pengkodean riwayat pengguna berurutan yang berbeda disediakan dengan basis kode ini:

  • Encoder bag-of-words (BOW): rata-rata embeddings aktivitas pengguna tanpa mempertimbangkan urutan konteks.
  • Encoder jaringan neural konvolusional (CNN): menerapkan beberapa lapisan jaringan neural konvolusional untuk menghasilkan encoding konteks.
  • Encoder jaringan neural berulang (RNN): menerapkan jaringan neural berulang untuk mengenkode urutan konteks.

Contoh

ID masukan:

  • Matriks (ID: 260)
  • Menyelamatkan Prajurit Ryan (ID: 2028)
  • (dan lebih)

ID Keluaran:

  • Star Wars: Episode VI - Kembalinya Jedi (ID: 1210)
  • (dan lebih)

Tolok ukur kinerja

Angka tolok ukur kinerja dibuat dengan alat yang dijelaskan di sini .

Nama model Ukuran Model Alat CPU
rekomendasi 0,52 Mb Pixel 3 0,09 md *
Pixel 4 0,05 md *

* 4 utas digunakan.

Gunakan data pelatihan Anda

Selain model terlatih, kami menyediakan toolkit open source di GitHub untuk melatih model dengan data Anda sendiri. Anda dapat mengikuti tutorial ini untuk mempelajari cara menggunakan toolkit dan menerapkan model terlatih di aplikasi seluler Anda sendiri.

Ikuti tutorial ini untuk menerapkan teknik yang sama yang digunakan di sini untuk melatih model rekomendasi menggunakan kumpulan data Anda sendiri.

Tips untuk penyesuaian model dengan data Anda

Model yang dilatih sebelumnya yang terintegrasi dalam aplikasi demo ini dilatih dengan kumpulan data MovieLens , Anda mungkin ingin mengubah konfigurasi model berdasarkan data Anda sendiri, seperti ukuran vocab, penyematan redup, dan panjang konteks masukan. Berikut beberapa tip:

  • Panjang konteks masukan: Panjang konteks masukan terbaik bervariasi dengan kumpulan data. Kami menyarankan memilih panjang konteks masukan berdasarkan seberapa banyak peristiwa label yang berkorelasi dengan kepentingan jangka panjang vs konteks jangka pendek.

  • Pemilihan jenis encoder: sebaiknya pilih jenis encoder berdasarkan panjang konteks masukan. Encoder bag-of-words bekerja dengan baik untuk panjang konteks input pendek (misalnya <10), encoder CNN dan RNN menghadirkan kemampuan peringkasan yang lebih untuk panjang konteks input yang panjang.