Python Util.grab_object Examples

Example #1

File: flaskServer.py Project: zhoulin886/python-work

def create_tree(offline):
    global model
    module_path = dirname(__file__)
    if offline:
       digits = load_data("video_report_201507_v5.txt")
       X = digits.data[:,1:]
       y = digits.target.astype(np.int64)
       model = DecisionTreeRegressor(max_depth=80)
       model.fit(X, y)
       Util.store_object(model,join(module_path, 'data', 'tree_model'))
    else:
        model = Util.grab_object(join(module_path, 'data', 'tree_model'))
    return model

Example #2

def create_tree(offline):
    global model
    module_path = dirname(__file__)
    if offline:
        digits = load_data("video_report_201507_v5.txt")
        X = digits.data[:, 1:]
        y = digits.target.astype(np.int64)
        model = DecisionTreeRegressor(max_depth=80)
        model.fit(X, y)
        Util.store_object(model, join(module_path, 'data', 'tree_model'))
    else:
        model = Util.grab_object(join(module_path, 'data', 'tree_model'))
    return model