Tạo câu trả lời thông minh bằng Bộ công cụ học máy trên iOS

Bạn có thể dùng Bộ công cụ học máy để tạo tin nhắn trả lời bằng mô hình trên thiết bị.

Để tạo các câu trả lời thông minh, bạn sẽ truyền cho Bộ công cụ học máy nhật ký các tin nhắn gần đây trong một cuộc trò chuyện. Nếu Bộ công cụ học máy xác định cuộc trò chuyện bằng tiếng Anh và cuộc trò chuyện không có chủ đề có thể nhạy cảm, thì Bộ công cụ học máy sẽ tạo tối đa 3 câu trả lời để bạn có thể đề xuất cho người dùng của mình.

Trước khi bắt đầu

  1. Nếu bạn chưa thêm Firebase vào ứng dụng của mình, hãy thực hiện bằng cách làm theo các bước trong hướng dẫn bắt đầu sử dụng.
  2. Đưa các thư viện Bộ công cụ học máy vào Podfile của bạn:
    pod 'Firebase/MLCommon', '6.25.0'
    pod 'Firebase/MLNLSmartReply', '6.25.0'
    
    Sau khi bạn cài đặt hoặc cập nhật Nhóm của dự án, hãy nhớ mở dự án Xcode bằng cách sử dụng .xcworkspace của dự án đó.
  3. Trong ứng dụng của bạn, hãy nhập Firebase:

    Swift

    import Firebase

    Objective-C

    @import Firebase;

1. Tạo một đối tượng nhật ký trò chuyện

Để tạo các câu trả lời thông minh, bạn hãy truyền cho Bộ công cụ học máy một mảng các đối tượng TextMessage theo thứ tự thời gian, trong đó dấu thời gian sớm nhất sẽ xuất hiện đầu tiên. Bất cứ khi nào người dùng gửi hoặc nhận tin nhắn, hãy thêm tin nhắn, dấu thời gian và mã nhận dạng người dùng của người gửi tin nhắn vào nhật ký trò chuyện.

Mã nhận dạng người dùng có thể là bất kỳ chuỗi nào xác định duy nhất người gửi trong cuộc trò chuyện. Mã nhận dạng người dùng không cần tương ứng với bất kỳ dữ liệu người dùng nào và mã nhận dạng người dùng không cần phải nhất quán giữa các cuộc trò chuyện hoặc lệnh gọi của trình tạo câu trả lời thông minh.

Nếu tin nhắn là do người dùng mà bạn muốn đề xuất câu trả lời gửi, hãy đặt isLocalUser thành true.

Swift

var conversation: [TextMessage] = []

// Then, for each message sent and received:
let message = TextMessage(
    text: "How are you?",
    timestamp: Date().timeIntervalSince1970,
    userID: "userId",
    isLocalUser: false)
conversation.append(message)

Objective-C

NSMutableArray *conversation = [NSMutableArray array];

// Then, for each message sent and received:
FIRTextMessage *message = [[FIRTextMessage alloc]
        initWithText:@"How are you?"
        timestamp:[NSDate date].timeIntervalSince1970
        userID:userId
        isLocalUser:NO];
[conversation addObject:message];

Đối tượng nhật ký trò chuyện sẽ có dạng như trong ví dụ sau:

Dấu thời gian User ID Bạn là người dùng cục bộ? Nội dung
Thứ Năm, ngày 21 tháng 2 13:13:39 theo giờ PST 2019 đúng bạn đang trên đường đi chưa?
Thứ Năm, ngày 21 tháng 2 13:15:03 giờ PST 2019 BẠN BÈ false Xin lỗi bạn vì muộn giờ!

Xin lưu ý rằng thông báo gần đây nhất trong ví dụ trên là của một người dùng không phải người dùng cục bộ. Việc này rất quan trọng vì Bộ công cụ học máy đề xuất các câu trả lời mà người dùng ứng dụng của bạn, tức là người dùng cục bộ, sẽ gửi đi. Bạn phải đảm bảo truyền cho Bộ công cụ học máy một nhật ký trò chuyện kết thúc bằng một thông báo mà người dùng của bạn có thể muốn trả lời.

2. Nhận tin nhắn trả lời

Để tạo câu trả lời thông minh cho một tin nhắn, hãy lấy một thực thể của SmartReply và chuyển nhật ký trò chuyện vào phương thức suggestReplies(for:completion:):

Swift

let naturalLanguage = NaturalLanguage.naturalLanguage()
naturalLanguage.smartReply().suggestReplies(for: conversation) { result, error in
    guard error == nil, let result = result else {
        return
    }
    if (result.status == .notSupportedLanguage) {
        // The conversation's language isn't supported, so the
        // the result doesn't contain any suggestions.
    } else if (result.status == .success) {
        // Successfully suggested smart replies.
        // ...
    }
}

Objective-C

FIRNaturalLanguage *naturalLanguage = [FIRNaturalLanguage naturalLanguage];
FIRSmartReply *smartReply = [naturalLanguage smartReply];
[smartReply suggestRepliesForMessages:inputText
                           completion:^(FIRSmartReplySuggestionResult * _Nullable result,
                                        NSError * _Nullable error) {
  if (error || !result) {
    return;
  }
  if (result.status == FIRSmartReplyResultStatusNotSupportedLanguage) {
      // The conversation's language isn't supported, so the
      // the result doesn't contain any suggestions.
  } else if (result.status == FIRSmartReplyResultStatusSuccess) {
      // Successfully suggested smart replies.
      // ...
  }
}];
]

Nếu thao tác thành công, đối tượng SmartReplySuggestionResult sẽ được chuyển đến trình xử lý hoàn thành. Đối tượng này chứa danh sách tối đa 3 câu trả lời đề xuất mà bạn có thể hiển thị cho người dùng của mình:

Swift

for suggestion in result.suggestions {
  print("Suggested reply: \(suggestion.text)")
}

Objective-C

for (FIRSmartReplySuggestion *suggestion in result.suggestions) {
  NSLog(@"Suggested reply: %@", suggestion.text);
}

Xin lưu ý rằng Bộ công cụ học máy có thể không trả về kết quả nếu mô hình không tự tin về mức độ liên quan của các câu trả lời đề xuất, cuộc trò chuyện đầu vào không bằng tiếng Anh hoặc nếu mô hình phát hiện thấy vấn đề nhạy cảm.