model_extension_snippet.swift

//
// MARK: - CoreML Methods
//

extension TablaViewController
{
    /**
        Predice el doble de un número usando un modelo `CoreML`
    
        - Parameter for: Número del que queremos averiguar su doble
        - Returns: El doble en caso que la predicción se haya realizado bien
    */
    private func predictResult(for number: Int) -> Int?
    {
        guard let prediction = try? model.prediction(multiplicador: Double(number)) else
        {
            return nil
        }

        return Int(prediction.resultado)
    }
}