Skip to content

Instantly share code, notes, and snippets.

@fitomad
Created January 19, 2018 18:50
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save fitomad/6c4e15f2d538674729943a5e7273f209 to your computer and use it in GitHub Desktop.
Save fitomad/6c4e15f2d538674729943a5e7273f209 to your computer and use it in GitHub Desktop.
Modelo que predice el doble de un valor dado. Usan progresión lineal para la predicción
from sklearn.linear_model import LinearRegression
# N veces
X_train = [[0], [1], [2], [3], [4], [5], [6], [7], [8], [9], [10]]
# Resultado
y_train = [0, 2, 4, 6, 8, 10, 12, 14, 16, 18, 20]
model = LinearRegression()
model.fit(X_train, y_train)
# Convertimos y guardamos el modelo
# de scikit-learn
import coremltools
coreml_model = coremltools.converters.sklearn.convert(model, ["multiplicador"], "resultado")
# Metadatos del modelo
coreml_model.author = 'Adolfo Vera'
coreml_model.license = 'BSD'
coreml_model.short_description = 'Predice el doble de un valor entero'
# Descripcion de los parametros de entrada
coreml_model.input_description['multiplicador'] = 'El valor del que queremos predecir el doble'
# Descripcion de los parametros de salida
coreml_model.output_description['resultado'] = 'El doble'
coreml_model.save('TablaModel.mlmodel')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment