Skip to content

Instantly share code, notes, and snippets.

@nickinack

nickinack/f12.py

Last active November 22, 2020 06:09
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 nickinack/08632259c39d6f326ee1353504311e72 to your computer and use it in GitHub Desktop.
Save nickinack/08632259c39d6f326ee1353504311e72 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
f12.py
def fitness_func(chromosome):
columns = []
for i in range(len(x_train.columns)):
if i in chromosome:
columns.append(x_train.columns[i])
dist.append(columns)
training_set = x_train[columns]
print(training_set)
test_set = x_test[columns]
lg = LinearRegression().fit(training_set.values, y_train.values)
preds = lg.predict(test_set.values)
return 100 / np.sqrt(mean_squared_error(y_test.values, preds))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment