Commit e91f03d8 authored by Laura Denies's avatar Laura Denies
Browse files

Upload New File

parent 778844d5
#!/usr/bin/env python
import pandas as pd
import numpy as np
import pickle
# Open file with pd.read_csv
df = pd.read_csv("matrix_model_complete.tsv", sep='\t')
X = df.drop('class', axis=1)
X = X.drop('#', axis=1)
y = df['class']
from sklearn.model_selection import train_test_split
# implementing train-test-split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.33, random_state=66)
from sklearn.ensemble import RandomForestClassifier
forest = RandomForestClassifier(n_estimators = 1600, max_depth=340, max_features='auto', oob_score = True), y_train)
print('Score: ', forest.score(X_train, y_train))
print('oob_score: ',forest.oob_score_)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment