Skip to content

Instantly share code, notes, and snippets.

@osherz
osherz / lmt.py
Last active December 16, 2020 11:57 — forked from logandillard/lmt.py
import math
import numpy as np
import pandas as pd
from sklearn.linear_model import Ridge
class LinearModelTree:
def __init__(self, min_node_size, node_model_fit_func, min_split_improvement=0):
self.min_node_size = min_node_size
self.node_model_fit_func = node_model_fit_func
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.