f20d62843c280765b984d3cc77f1eb7f61d58c4f,mltsp/build_model.py,,create_and_pickle_model,#,187

Before Change

    
    // Fit the model
    rf_fit_pkl = fit_model.delay(data_dict)
    rf_fit_pkl = rf_fit_pkl.get(timeout=20)
    rf_fit = pickle.loads(rf_fit_pkl)

    // Store the model:
    print("Pickling model...")

After Change

    // Fit the model to training data:
    print("Fitting the model...")
    rf_fit.fit(data_dict["features"], data_dict["classes"])
    print("Done.")
    del data_dict

    // Store the model:
    print("Pickling model...")

Instances