sololearn学习笔记Machine Learning-17

Building a Logistic Regression model.

任务

Task
You are given a feature matrix and a single datapoint to predict. Your job will be to build a Logistic Regression model with the feature matrix and make a prediction (1 or 0) of the single datapoint.

输入格式

First line: Number of data points in the feature matrix (n)
Next n lines: Values of the row in the feature matrix, separated by spaces
Next line: Target values separated by spaces
Final line: Values (separated by spaces) of a single datapoint without a target value

输出格式

Output Format
Either 1 or 0

输入案例

Sample Input
6
1 3
3 5
5 7
3 1
5 3
7 5
1 1 1 0 0 0
2 4

输出案例

Sample Output
1

代码

from sklearn.linear_model import LogisticRegression
import numpy as np
n = int(input())
X = []
for i in range(n):
    a = [float(x) for x in input().split()]
    X.append(a)
label = [int(x) for x in input().split()]
model = LogisticRegression(C=50)
X1 = [float(x) for x in input().split()]
X1 = np.reshape(X1, [1, -1])
model.fit(X, label)
predict = model.predict(X1)
print(predict[0])